Update app.py

app.py

@@ -2,13 +2,23 @@ import gradio as gr
 import requests
 
 # 设置 Ollama API 地址
-OLLAMA_API_URL = "http://localhost:7860"  # 这个是你本地 Ollama 服务的 URL
+OLLAMA_API_URL = "http://localhost:7860/api/query"  # 替换为实际路径
 
 # 定义请求 Ollama API 的函数
 def get_ollama_response(query):
-    # 发送请求到 Ollama API
-    response = requests.post(OLLAMA_API_URL, json={"query": query})
-    return response.json()["answer"]  # 假设返回的 JSON 格式包含 "answer"
+    try:
+        # 发送请求到 Ollama API
+        response = requests.post(OLLAMA_API_URL, json={"query": query})
+        if response.status_code != 200:
+            return f"Error: Server returned status code {response.status_code}"
+        
+        # 解析 JSON 响应
+        response_data = response.json()
+        print("API Response:", response_data)  # 调试时查看完整响应
+        return response_data.get("answer", "No answer found in response.")
+    except Exception as e:
+        print("Error:", e)
+        return "Error: Unable to process the request."
 
 # 使用 Gradio 创建界面
 iface = gr.Interface(fn=get_ollama_response, inputs="text", outputs="text")