Update app.py

app.py

@@ -13,35 +13,63 @@ def query_vectara(question):
     CORPUS_ID = config('CORPUS_ID')
     API_KEY = config('API_KEY')
 
-    # Define the data dictionary
-    data_dict = {
+    # Define the headers
+    api_key_header = {
+        "customer-id": CUSTOMER_ID,
+        "x-api-key": API_KEY
+    }
+
+    # Define the request body in the structure provided in the example
+    request_body = {
         "query": [
             {
                 "query": user_message,
-                "num_results": 10,
-                "corpus_key": [
+                "queryContext": "",
+                "start": 0,
+                "numResults": 25,
+                "contextConfig": {
+                    "charsBefore": 0,
+                    "charsAfter": 0,
+                    "sentencesBefore": 2,
+                    "sentencesAfter": 2,
+                    "startTag": "%START_SNIPPET%",
+                    "endTag": "%END_SNIPPET%",
+                },
+                "rerankingConfig": {
+                    "rerankerId": 272725718,
+                    "mmrConfig": {
+                        "diversityBias": 0.3
+                    }
+                },
+                "corpusKey": [
                     {
-                        "customer_id": CUSTOMER_ID,
-                        "corpus_id": CORPUS_ID
+                        "customerId": CUSTOMER_ID,
+                        "corpusId": CORPUS_ID,
+                        "semantics": 0,
+                        "metadataFilter": "",
+                        "lexicalInterpolationConfig": {
+                            "lambda": 0
+                        },
+                        "dim": []
+                    }
+                ],
+                "summary": [
+                    {
+                        "maxSummarizedResults": 1,
+                        "responseLang": "eng",
+                        "summarizerPromptName": "vectara-summary-ext-v1.2.0"
                     }
                 ]
             }
         ]
     }
 
-    # Define the headers
-    api_key_header = {
-        "customer-id": CUSTOMER_ID,
-        "x-api-key": API_KEY,
-        "Content-Type": "application/json"  # Set Content-Type header
-    }
-
-    # Make the API request
+    # Make the API request using Gradio
     response = requests.post(
         "https://api.vectara.io/v1/query",
-        data=json.dumps(data_dict),
+        json=request_body,  # Use json to automatically serialize the request body
         verify=True,
-        headers=api_key_header  # Use the updated headers dictionary
+        headers=api_key_header
     )
 
     if response.status_code == 200:
@@ -55,7 +83,7 @@ def query_vectara(question):
 iface = gr.Interface(
     fn=query_vectara,
     inputs=[gr.Textbox(label="Input Text")],
-    outputs=[gr.Textbox(label="Output Text")],
+    outputs=gr.Textbox(label="Output Text"),
     title="Vectara Chatbot",
     description="Ask me anything using the Vectara API!"
 )