query.py

import requests
import json

def query_vectara(question, customer_id, api_key, corpus_key):
    """Queries the uploaded documents in Vectara API v2."""
    url = f"https://api.vectara.io/v2/corpora/{corpus_key}/query"
    headers = {
        "x-api-key": api_key,
        "Accept": "application/json"
    }
    
    payload = {
        "query": question,
        "top_k": 5  # Retrieve top 5 relevant results
    }
    
    response = requests.get(url, headers=headers, data=json.dumps(payload))
    return response.json()

def process_queries(customer_id, api_key, corpus_key):
    """Runs all predefined queries on the uploaded documents."""
    questions = [
        "Based on uploaded documents, what are the top four challenges of the Fintech sector in Saudi Arabia? list them in bullet points.",
        "Based on uploaded documents, who are the top five leading companies in the Fintech sector in Saudi Arabia? list them in bullet points with brief information on each company.",
        "Based on uploaded documents, summarize Saudi Arabia's Fintech Strategy in five key points.",
        "Based on uploaded documents, describe STC Pay, its position in the market, and partners.",
        "Based on uploaded documents, what are the key differences between Fintech in Saudi vs UAE?",
        "Based on uploaded documents, summarize the Payment Services Provider Regulations (PSPR) in five bullet points."
    ]
    
    results = {}
    for question in questions:
        response = query_vectara(question, customer_id, api_key, corpus_key)
        results[question] = response
    
    return results