Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,36 +3,15 @@ import requests
|
|
| 3 |
import json
|
| 4 |
from decouple import Config
|
| 5 |
|
| 6 |
-
def query_vectara(question, chat_history
|
| 7 |
-
# Handle file upload to Vectara if a file is provided
|
| 8 |
-
if uploaded_file is not None:
|
| 9 |
-
customer_id = config('CUSTOMER_ID') # Read from .env file
|
| 10 |
-
corpus_id = config('CORPUS_ID') # Read from .env file
|
| 11 |
-
api_key = config('API_KEY') # Read from .env file
|
| 12 |
-
url = f"https://api.vectara.io/v1/upload?c={customer_id}&o={corpus_id}"
|
| 13 |
-
|
| 14 |
-
post_headers = {
|
| 15 |
-
"x-api-key": api_key,
|
| 16 |
-
"customer-id": customer_id
|
| 17 |
-
}
|
| 18 |
-
|
| 19 |
-
files = {
|
| 20 |
-
"file": (uploaded_file.name, uploaded_file),
|
| 21 |
-
"doc_metadata": (None, json.dumps({"metadata_key": "metadata_value"})), # Replace with your metadata
|
| 22 |
-
}
|
| 23 |
-
response = requests.post(url, files=files, headers=post_headers)
|
| 24 |
-
|
| 25 |
-
if response.status_code == 200:
|
| 26 |
-
upload_status = "File uploaded successfully"
|
| 27 |
-
else:
|
| 28 |
-
upload_status = "Failed to upload the file"
|
| 29 |
-
else:
|
| 30 |
-
upload_status = "No file uploaded"
|
| 31 |
-
|
| 32 |
# Get the user's message from the chat history
|
| 33 |
user_message = chat_history[-1][0]
|
| 34 |
|
| 35 |
# Query Vectara API
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
query_url = "https://api.vectara.io/v1/query/v1/query"
|
| 37 |
|
| 38 |
headers = {
|
|
@@ -89,16 +68,16 @@ def query_vectara(question, chat_history, uploaded_file=None):
|
|
| 89 |
|
| 90 |
if query_response.status_code == 200:
|
| 91 |
query_data = query_response.json()
|
| 92 |
-
response_message = f"
|
| 93 |
else:
|
| 94 |
-
response_message = f"
|
| 95 |
|
| 96 |
return response_message
|
| 97 |
|
| 98 |
-
# Create a Gradio ChatInterface with a text input
|
| 99 |
iface = gr.Interface(
|
| 100 |
fn=query_vectara,
|
| 101 |
-
inputs=[gr.Textbox(label="Input Text")
|
| 102 |
outputs=gr.Textbox(label="Output Text"),
|
| 103 |
title="Vectara Chatbot",
|
| 104 |
description="Ask me anything using the Vectara API!"
|
|
|
|
| 3 |
import json
|
| 4 |
from decouple import Config
|
| 5 |
|
| 6 |
+
def query_vectara(question, chat_history):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
# Get the user's message from the chat history
|
| 8 |
user_message = chat_history[-1][0]
|
| 9 |
|
| 10 |
# Query Vectara API
|
| 11 |
+
customer_id = config('CUSTOMER_ID') # Read from .env file
|
| 12 |
+
corpus_id = config('CORPUS_ID') # Read from .env file
|
| 13 |
+
api_key = config('API_KEY') # Read from .env file
|
| 14 |
+
|
| 15 |
query_url = "https://api.vectara.io/v1/query/v1/query"
|
| 16 |
|
| 17 |
headers = {
|
|
|
|
| 68 |
|
| 69 |
if query_response.status_code == 200:
|
| 70 |
query_data = query_response.json()
|
| 71 |
+
response_message = f"Response from Vectara API: {json.dumps(query_data, indent=2)}"
|
| 72 |
else:
|
| 73 |
+
response_message = f"Error: {query_response.status_code}"
|
| 74 |
|
| 75 |
return response_message
|
| 76 |
|
| 77 |
+
# Create a Gradio ChatInterface with only a text input
|
| 78 |
iface = gr.Interface(
|
| 79 |
fn=query_vectara,
|
| 80 |
+
inputs=[gr.Textbox(label="Input Text")],
|
| 81 |
outputs=gr.Textbox(label="Output Text"),
|
| 82 |
title="Vectara Chatbot",
|
| 83 |
description="Ask me anything using the Vectara API!"
|