Update app.py

app.py

@@ -4,12 +4,12 @@ import logging
 from typing import List, Dict
 import torch
 import gradio as gr
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 from langchain.prompts import PromptTemplate
-from langchain.llms import HuggingFacePipeline
+from langchain_community.llms import HuggingFacePipeline
 from langchain_community.document_loaders import PyPDFLoader, TextLoader, Docx2txtLoader
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 
@@ -246,46 +246,36 @@ def process_query(message, history):
         return history + [(message, f"Error: {str(e)}")]
 
 # Create Gradio interface
-demo = gr.Blocks(css="""
-    div.gradio-container {background-color: #f0f2f6}
-    div.upload-box { 
-        border: 2px dashed #ccc;
-        padding: 20px;
-        border-radius: 10px;
-        background-color: #ffffff;
-        margin-bottom: 15px;
-    }
-""")
-
-with demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.HTML("""
-        <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
+        <div style="text-align: center; margin-bottom: 1rem;">
             <h1 style="color: #2d333a;">🤖 Easy RAG</h1>
-            <p style="color: #4a5568;">
-                A simple and powerful RAG system for your documents
-            </p>
+            <p style="color: #4a5568;">A simple and powerful RAG system for your documents</p>
         </div>
     """)
     
     with gr.Row():
         # Sidebar for document upload
         with gr.Column(scale=1):
-            with gr.Box(elem_classes="upload-box"):
+            with gr.Box():
                 gr.HTML("<h3>📁 Document Upload</h3>")
                 file_output = gr.File(
                     file_count="multiple",
-                    label="Upload Documents (PDF, TXT, DOCX)",
-                    elem_classes="file-upload"
+                    label="Upload Documents",
+                    elem_id="file-upload"
                 )
                 gr.HTML("""
                     <div style="font-size: 0.8em; color: #666;">
                         <p>• Maximum 5 files</p>
                         <p>• 10MB per file</p>
-                        <p>• Supported formats: PDF, TXT, DOCX</p>
+                        <p>• Supported: PDF, TXT, DOCX</p>
                     </div>
                 """)
                 upload_button = gr.Button("📤 Upload and Initialize", variant="primary")
-                system_output = gr.Textbox(label="System Status", interactive=False)
+                system_output = gr.Textbox(
+                    label="System Status",
+                    interactive=False
+                )
         
         # Main chat area
         with gr.Column(scale=3):
@@ -303,24 +293,24 @@ with demo:
                     container=False,
                     scale=8
                 )
-                clear = gr.Button("🗑️ Clear", size="sm", scale=1)
+                clear = gr.Button("🗑️", size="sm", scale=1)
     
     gr.HTML("""
-        <div style="text-align: center; max-width: 800px; margin: 20px auto; padding: 20px;
+        <div style="text-align: center; max-width: 800px; margin: 20px auto; padding: 1rem;
                     background-color: #f8f9fa; border-radius: 10px;">
-            <div style="margin-bottom: 15px;">
+            <div style="margin-bottom: 1rem;">
                 <h3 style="color: #2d333a;">🔍 About Easy RAG</h3>
-                <p style="color: #666; font-size: 14px;">
-                    A powerful RAG system that lets you query your documents using:
+                <p style="color: #666; font-size: 0.9em;">
+                    Powered by state-of-the-art AI technology:
                 </p>
-                <ul style="list-style: none; color: #666; font-size: 14px;">
+                <ul style="list-style: none; color: #666; font-size: 0.9em;">
                     <li>🔹 LLM: Llama-2-7b-chat-hf</li>
                     <li>🔹 Embeddings: multilingual-e5-large</li>
                     <li>🔹 Vector Store: FAISS</li>
                 </ul>
             </div>
-            <div style="border-top: 1px solid #ddd; padding-top: 15px;">
-                <p style="color: #666; font-size: 14px;">
+            <div style="border-top: 1px solid #ddd; padding-top: 1rem;">
+                <p style="color: #666; font-size: 0.8em;">
                     Based on original work by <a href="https://www.linkedin.com/in/camilo-vega-169084b1/" 
                     target="_blank" style="color: #2196F3; text-decoration: none;">Camilo Vega</a>
                 </p>
@@ -343,4 +333,5 @@ with demo:
     
     clear.click(lambda: None, None, chatbot)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()