Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import tempfile
+import torch
 import gradio as gr
 from langchain_community.vectorstores import FAISS
 from langchain_groq import ChatGroq
@@ -10,14 +11,17 @@ from langchain.document_loaders import PyPDFLoader
 from langchain import hub
 
 # Set API key (Replace with your actual key)
-os.environ["GROQ_API_KEY"] = "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"
+os.environ["GROQ_API_KEY"] = "your_groq_api_key"
 
-# Initialize LLM and Embeddings
+# Check if GPU is available
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Initialize LLM and Embeddings with GPU if available
 llm = ChatGroq(model="llama3-8b-8192")
 model_name = "BAAI/bge-small-en"
 hf_embeddings = HuggingFaceBgeEmbeddings(
     model_name=model_name,
-    model_kwargs={'device': 'cpu'},
+    model_kwargs={'device': device},
     encode_kwargs={'normalize_embeddings': True}
 )
 
@@ -64,23 +68,28 @@ def ask_question(query):
     if "rag_chain" not in globals():
         return "Please upload and process a PDF first."
     
-    response = rag_chain.invoke(query).content
+    response = rag_chain.invoke(query)
     return response
 
-# Gradio UI
+# Gradio UI with Mobile-Friendly Fixes
 with gr.Blocks() as demo:
     gr.Markdown("# 📄 PDF Chatbot with RAG")
     gr.Markdown("Upload a PDF and ask questions!")
-    
-    pdf_input = gr.File(label="Upload PDF", type="binary")
-    process_button = gr.Button("Process PDF")
+
+    with gr.Row():
+        pdf_input = gr.File(label="Upload PDF", type="binary")
+        process_button = gr.Button("Process PDF")
+
     output_message = gr.Textbox(label="Status", interactive=False)
-    
-    query_input = gr.Textbox(label="Ask a Question")
-    submit_button = gr.Button("Submit")
+
+    with gr.Row():
+        query_input = gr.Textbox(label="Ask a Question")
+        submit_button = gr.Button("Submit")
+
     response_output = gr.Textbox(label="AI Response")
 
     process_button.click(process_pdf, inputs=pdf_input, outputs=output_message)
     submit_button.click(ask_question, inputs=query_input, outputs=response_output)
 
-demo.launch()
+# Improved mobile layout and timeout handling
+demo.launch(share=True, theme="compact", show_api=False, timeout=60)