Update app.py

app.py

@@ -6,28 +6,28 @@ from RAGModule import RAGModule
 # Initialize the RAG module
 RAG_Triwira = RAGModule()
 
-# def chat(message: str, chat_history: str):
-#     prompt = RAG_Triwira.main(message)
-#     url = "http://localhost:11434/api/generate"
-#     payload = json.dumps({
-#         "model": "MarcoAland/llama3.1-rag-indo",
-#         "prompt": prompt,
-#         "format": "json",
-#         "stream": False
-#     })
-#     headers = {
-#         'Content-Type': 'application/json'
-#     }
+def chat(message: str, chat_history: str):
+    prompt = RAG_Triwira.main(message)
+    url = "http://localhost:11434/api/generate"
+    payload = json.dumps({
+        "model": "ollama:latest", #"MarcoAland/llama3.1-rag-indo",
+        "prompt": prompt,
+        "format": "json",
+        "stream": False
+    })
+    headers = {
+        'Content-Type': 'application/json'
+    }
 
-#     try:
-#         response = requests.post(url, headers=headers, data=payload)
-#         response.raise_for_status()  # Raise an error for bad status codes
-#         response_data = response.json()
-#         return response_data.get('output', 'No response received')
-#     except requests.exceptions.RequestException as e:
-#         return f"Connection error: {e}"
-def chat(message, chat_history):
-    yield "test"
+    try:
+        response = requests.post(url, headers=headers, data=payload)
+        response.raise_for_status()  # Raise an error for bad status codes
+        response_data = response.json()
+        return response_data.get('output', 'No response received')
+    except requests.exceptions.RequestException as e:
+        return f"Connection error: {e}"
+# def chat(message, chat_history):
+#     yield "test"
 # Define the Gradio interface
 iface = gr.Interface(
     fn=chat,