Update app.py

app.py

@@ -49,30 +49,45 @@ def query_with_retry(query, max_retries=3, wait_time=5):
             break
 
 # Kullanıcı mesajlarını ve botun cevabını yönetmek
-def answer(message, history):
-    bot_response = query_with_retry(message)
+def respond(message, history):
+    try:
+        # Sorguyu query_engine üzerinde çalıştır
+        response = query_engine.query(message)
+        
+        # Loglama işlemi
+        print(f"\n{datetime.now()}:{llm.model_name}:: {message} --> {str(response)}\n")
+        
+        return f"{llm.model_name}:\n{str(response)}"
+    except Exception as e:
+        if str(e) == "'NoneType' object has no attribute 'as_query_engine'":
+            return "Please upload a file."
+        return f"An error occurred: {e}"
+# UI Setup
+with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto Mono")]), css='footer {visibility: hidden}') as demo:
+    gr.Markdown("# DocBot📄🤖")
     
-    # Eğer bot cevabından metin almak gerekiyorsa, .text ile metin almak gerekiyor.
-    bot_text = bot_response.text if hasattr(bot_response, 'text') else str(bot_response)
-    
-    # Kullanıcı mesajını ve bot cevabını ekleyelim
-    history.append({"role": "user", "content": message})
-    history.append({"role": "bot", "content": bot_text.strip()})
-    
-    return history
+    with gr.Tabs():
+        with gr.TabItem("Intro"):
+            gr.Markdown("## Welcome to DocBot! 📄🤖\nAsk questions related to the uploaded document.")
 
-# Gradio Chat arayüzünü oluşturma
-def create_interface():
-    # Gradio arayüzü oluşturma
-    interface = gr.ChatInterface(
-        fn=answer,
-        type="messages",
-        title="Hund Ecosystem Question Answering",
-        description="Ask questions based on the Hund Ecosystem PDF document you uploaded.",
-    )
-    return interface
+        with gr.TabItem("DocBot"):
+            with gr.Accordion("=== IMPORTANT: READ ME FIRST ===", open=False):
+                gr.Markdown("## Please ask questions about the uploaded document.\nYou can interact with the document once it's loaded.")
+            
+            with gr.Row():
+                with gr.Column(scale=3):
+                    gr.ChatInterface(
+                        fn=respond,  # `respond` fonksiyonu, kullanıcı sorularını cevaplamak için kullanılacak
+                        chatbot=gr.Chatbot(height=500),
+                        theme="soft",
+                        show_progress='full',
+                        textbox=gr.Textbox(placeholder="Step-1: Ask me questions about the uploaded document!", container=False)
+                    )
+                
+    # Set up Gradio interactions (Gerekli işlevlerin burada çalıştığından emin olun)
+    btn.click(fn=load_files, inputs=[file_input], outputs=output)
+    clear.click(lambda: [None] * 1, outputs=[file_input])  # Yalnızca dosya yükleme temizlenecek
 
-# Çalıştırma
+# Launch the demo
 if __name__ == "__main__":
-    # Arayüzü başlatıyoruz
-    create_interface().launch()
+    demo.launch()