Update app.py

app.py

@@ -1,86 +1,101 @@
 import gradio as gr
-import os
-from typing import List
-from datetime import datetime
 
-uploaded_files = []
+def process_question(question, history, uploaded_files):
+    # This is where you would integrate your RAG system logic.
+    # For this example, we'll just return a placeholder response.
+    if uploaded_files is None:
+        return history, [("", "Please upload a PDF first.")]
+    answer = f"Answer to: {question} (Based on {len(uploaded_files)} uploaded files)"
+    history.append((question, answer))
+    return history, [("", answer)]
 
-def upload_pdfs(files: List[gr.File]):
-    """
-    Saves the uploaded PDF files to a directory and tracks the uploaded files.
-    """
-    global uploaded_files
-    saved_files = []
-    try:
-        os.makedirs("uploaded_pdfs", exist_ok=True)
-        for file in files:
-            if file.name.endswith(".pdf"):
-                file_path = os.path.join("uploaded_pdfs", file.name)
-                # Write the file content to the destination
-                with open(file_path, "wb") as f:
-                    f.write(file.read())
-                saved_files.append(file.name)
-                if file.name not in uploaded_files:
-                    uploaded_files.append(file.name)
-            else:
-                return f"Error: {file.name} is not a PDF file."
-        if saved_files:
-            return f"Successfully uploaded: {', '.join(saved_files)}."
+with gr.Blocks(css="""
+    body {
+        background-color: #f4f4f4;
+    }
+    .gradio-container {
+        border-radius: 10px;
+        box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
+    }
+    .container {
+        padding: 20px;
+    }
+    .question-area {
+        margin-bottom: 20px;
+    }
+    .history-button {
+        background-color: #4CAF50;
+        color: white;
+        padding: 10px 20px;
+        border: none;
+        border-radius: 5px;
+        cursor: pointer;
+    }
+    .history-button:hover {
+        background-color: #46a049;
+    }
+    .gradio-chatbot > .wrap > div:last-child > div:last-child {
+        display: none !important;
+    }
+""") as demo:
+    gr.Markdown("## RAG System Demo")
+
+    with gr.Tabs() as tabs:
+        with gr.TabItem("Upload PDFs") as upload_tab:
+            uploaded_files = gr.Files(label="Upload at least one PDF", type="filepath")
+            with gr.Row():
+                upload_button = gr.Button("Upload")  # Upload button is visible initially
+                upload_status = gr.Textbox(visible=False)
+
+        with gr.TabItem("Chat", visible=False) as chat_tab:
+            question = gr.Textbox(label="Ask a Question", placeholder="Enter your question here...")
+            submit_btn = gr.Button("Submit")
+            rag_reply = gr.Chatbot(label="RAG Reply")
+
+        with gr.TabItem("History"):
+            history_markdown = gr.Markdown("No chat history yet.")
+
+    chat_history = gr.State([])
+
+    def activate_chat_tab():
+        return gr.Tabs.update(selected=1)
+
+    def upload_pdfs(files):
+        # Simulate an upload process (replace with your actual logic)
+        # time.sleep(2)
+        if files:
+            return gr.update(visible=True, value="Upload finished!"), gr.update(visible=True)
         else:
-            return "No PDF files were uploaded."
-    except Exception as e:
-        return f"Error uploading files: {str(e)}"
+            return gr.update(visible=True, value="Please upload at least one PDF file."), gr.update(visible=False)
+
+    upload_button.click(
+        upload_pdfs,
+        inputs=uploaded_files,
+        outputs=[upload_status, chat_tab],
+    )
 
-chat_history = []
+    upload_button.click(
+        fn=activate_chat_tab,
+        inputs=None,
+        outputs=tabs,
+    )
 
-def chat_with_rag(question: str):
-    """
-    Interacts with the RAG system and returns the answer.
-    """
-    global chat_history
-    # Placeholder response. Replace this with actual logic to interact with RAG system.
-    answer = f"The answer to your question '{question}' is: This is a placeholder response."
-    chat_history.append(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Q: {question}\nA: {answer}")
-    return answer
+    upload_status.change(
+        fn=lambda: gr.update(visible=False),
+        inputs=None,
+        outputs=[upload_status],
+    )
 
-def show_chat_history():
-    """
-    Displays the chat history.
-    """
-    global chat_history
-    if chat_history:
-        return "\n\n".join(chat_history)
-    else:
-        return "No chat history available."
+    submit_btn.click(
+        fn=process_question,
+        inputs=[question, chat_history, uploaded_files],
+        outputs=[chat_history, rag_reply],
+    )
 
-def create_gradio_interface():
-    with gr.Blocks(title="RAG System UI", theme=gr.themes.Soft()) as demo:
-        gr.Markdown("## RAG System Interface")
-        
-        with gr.Tab("File Upload"):
-            gr.Markdown("### Upload PDF Files")
-            file_uploader = gr.File(label="Upload PDFs", file_count="multiple")
-            upload_button = gr.Button("Upload PDFs")
-            upload_status = gr.Textbox(label="Upload Status", interactive=False)
-            upload_button.click(upload_pdfs, inputs=file_uploader, outputs=upload_status)
-            gr.Markdown("#### Uploaded Files")
-            uploaded_files_list = gr.Textbox(label="Uploaded Files", interactive=False)
-            uploaded_files_list.update(value=lambda: "\n".join(uploaded_files))
-        
-        with gr.Tab("Chat"):
-            gr.Markdown("### Ask a Question")
-            chat_input = gr.Textbox(label="Your Question")
-            chat_button = gr.Button("Get Answer")
-            chat_output = gr.Textbox(label="Answer", interactive=False)
-            chat_button.click(chat_with_rag, inputs=chat_input, outputs=chat_output)
-        
-        with gr.Tab("History"):
-            gr.Markdown("### Chat History")
-            history_button = gr.Button("Show Chat History")
-            history_output = gr.Textbox(label="Chat History", interactive=False)
-            history_button.click(show_chat_history, outputs=history_output)
-        
-    return demo
+    chat_history.change(
+        fn=lambda x: gr.Markdown("<br>".join([f"**Question:** {q}<br>**Answer:** {a}" for q, a in x])),
+        inputs=chat_history,
+        outputs=history_markdown,
+    )
 
-if __name__ == "__main__":
-    create_gradio_interface().launch()
+demo.launch()