Update app.py

app.py

@@ -18,6 +18,7 @@ try:
     GROQ_OK = True
 except ImportError:
     GROQ_OK = False
+    print("❌ Groq library not installed!")
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -31,20 +32,34 @@ groq_client = None
 if GROQ_OK:
     try:
         print("DEBUG → Initializing Groq client...")
-        groq_client = Groq(api_key=GROQ_API_KEY)  # <- only api_key
-        print("DEBUG → Groq client initialized successfully!")
+        # Initialize with just api_key - most compatible approach
+        groq_client = Groq(api_key=GROQ_API_KEY)
+        print("✅ DEBUG → Groq client initialized successfully!")
+    except TypeError as te:
+        # Fallback for version compatibility issues
+        print(f"⚠️ TypeError during init: {te}")
+        try:
+            print("🔄 Attempting fallback initialization...")
+            groq_client = Groq(api_key=GROQ_API_KEY)
+            print("✅ Fallback initialization successful!")
+        except Exception as e:
+            groq_client = None
+            print(f"❌ Groq initialization failed: {e}")
     except Exception as e:
         groq_client = None
-        print("❌ Groq initialization error:", e)
+        print(f"❌ Groq initialization error: {e}")
+else:
+    print("❌ Groq library import failed!")
 
 class AgenticRAGAgent:
     def __init__(self):
         self.chunks = []
         self.index = None
         self.embedder = SentenceTransformer('all-MiniLM-L6-v2')
+        print("✅ AgenticRAGAgent initialized with SentenceTransformer")
 
-    # Remove emojis completely
     def remove_emojis(self, text: str) -> str:
+        """Remove emojis from text for clean voice output"""
         emoji_pattern = re.compile("[" 
             u"\U0001F600-\U0001F64F" 
             u"\U0001F300-\U0001F5FF" 
@@ -56,19 +71,21 @@ class AgenticRAGAgent:
         return emoji_pattern.sub(r'', text)
 
     def clean_for_voice(self, text: str) -> str:
+        """Clean text for voice synthesis"""
         text = self.remove_emojis(text)
         text = re.sub(r'[\*_`#\[\]]', '', text)
         text = re.sub(r'\s+', ' ', text).strip()
         return text
 
     def generate_voice(self, text: str):
+        """Generate voice output from text"""
         if not text or not text.strip():
             return None
         clean = self.clean_for_voice(text)
         if len(clean) < 5:
             return None
         try:
-            tts = gTTS(text=clean, lang='en')
+            tts = gTTS(text=clean, lang='en', slow=False)
             tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
             tts.save(tmp.name)
             return tmp.name
@@ -77,6 +94,7 @@ class AgenticRAGAgent:
             return None
 
     def upload_pdfs(self, files):
+        """Upload and process PDF files"""
         if not files:
             return "No files selected."
 
@@ -86,14 +104,17 @@ class AgenticRAGAgent:
         count = 0
 
         for file in files:
-            if not str(file.name).lower().endswith('.pdf'):
+            filename = str(file.name) if hasattr(file, 'name') else str(file)
+            if not filename.lower().endswith('.pdf'):
                 continue
-            dest = folder / Path(file.name).name
+            
+            dest = folder / Path(filename).name
             try:
-                content = file.read() if hasattr(file, 'read') else open(file.name, 'rb').read()
+                content = file.read() if hasattr(file, 'read') else open(filename, 'rb').read()
                 with open(dest, "wb") as f:
                     f.write(content)
             except Exception as e:
+                logger.warning(f"Failed to save file {filename}: {e}")
                 continue
 
             text = ""
@@ -104,7 +125,8 @@ class AgenticRAGAgent:
                         t = page.extract_text()
                         if t:
                             text += t + " "
-            except:
+            except Exception as e:
+                logger.warning(f"Failed to extract text from {filename}: {e}")
                 continue
 
             if text.strip():
@@ -115,33 +137,43 @@ class AgenticRAGAgent:
         if not all_chunks:
             return "No readable text found in the PDFs."
 
-        vecs = self.embedder.encode([c["content"] for c in all_chunks], show_progress_bar=False)
+        # Create embeddings and FAISS index
+        print(f"Creating embeddings for {len(all_chunks)} chunks...")
+        vecs = self.embedder.encode([c["content"] for c in all_chunks], show_progress_bar=True)
         vecs = vecs / np.linalg.norm(vecs, axis=1, keepdims=True)
         dim = vecs.shape[1]
+        
         self.index = faiss.IndexFlatIP(dim)
         self.index.add(vecs.astype('float32'))
         self.chunks = all_chunks
 
-        return f"Loaded {count} PDF(s) → {len(all_chunks)} chunks ready!"
+        status_msg = f"✅ Loaded {count} PDF(s) → {len(all_chunks)} chunks ready!"
+        print(status_msg)
+        return status_msg
 
     def ask(self, question: str, history: List):
+        """Process user question and generate response"""
         global groq_client
+        
         if not question.strip():
             return history, None
 
         if not history:
             history = []
 
+        # Handle greeting
         if question.strip().lower() in ["hi", "hello", "hey", "hola", "howdy"]:
             reply = "Hi there! I am AI Research Agent with agentic capabilities. Upload PDF documents and ask complex questions!"
             history.append([question, reply])
             return history, self.generate_voice(reply)
 
+        # Check if PDFs are loaded
         if not self.index:
             reply = "Please upload a PDF document first!"
             history.append([question, reply])
             return history, self.generate_voice(reply)
 
+        # Retrieve relevant chunks
         q_vec = self.embedder.encode([question])
         q_vec = q_vec / np.linalg.norm(q_vec)
         D, I = self.index.search(q_vec.astype('float32'), k=6)
@@ -150,9 +182,11 @@ class AgenticRAGAgent:
         prompt = f"Context from documents:\n{context}\n\nQuestion: {question}\nAnswer clearly and accurately:"
 
         if groq_client is None:
-            reply = "GROQ_API_KEY is missing or invalid."
+            reply = "ERROR: Groq client is not initialized. Check your API key and connection."
+            print("❌ Groq client is None - cannot process request")
         else:
             try:
+                print(f"📤 Sending request to Groq API for question: {question[:50]}...")
                 resp = groq_client.chat.completions.create(
                     model="llama-3.3-70b-versatile",
                     messages=[{"role": "user", "content": prompt}],
@@ -160,56 +194,108 @@ class AgenticRAGAgent:
                     max_tokens=700
                 )
                 reply = resp.choices[0].message.content.strip()
+                print(f"✅ Received response from Groq API")
             except Exception as e:
                 reply = f"Groq API error: {str(e)}"
+                print(f"❌ Groq API error: {e}")
 
         history.append([question, reply])
         return history, self.generate_voice(reply)
 
+
 # =========================================
 # GRADIO UI
 # =========================================
 def create_interface():
     agent = AgenticRAGAgent()
 
-    with gr.Blocks(title="🤖 AI Research Agent", theme=gr.themes.Soft()) as interface:
+    with gr.Blocks(title="AI Research Agent", theme=gr.themes.Soft()) as interface:
         gr.HTML("""
         <div style="text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 15px;">
-            <h1 style="color: white; margin: 0;">🤖 AI Research Agent - Agentic RAG</h1>
-            <p style="color: white; margin: 10px 0;">Advanced Multi-Tool Research Assistant with Voice Support 🔊</p>
+            <h1 style="color: white; margin: 0;">AI Research Agent - Agentic RAG</h1>
+            <p style="color: white; margin: 10px 0;">Advanced Multi-Tool Research Assistant with Voice Support</p>
         </div>
         """)
 
         with gr.Row():
             with gr.Column(scale=2):
-                chatbot = gr.Chatbot(label="💬 Chat", height=500)
+                chatbot = gr.Chatbot(
+                    label="Chat",
+                    height=500,
+                    type="tuples"
+                )
 
                 with gr.Row():
-                    msg = gr.Textbox(label="", placeholder="Ask a complex research question...", scale=4)
-                    submit_btn = gr.Button("🚀 Send", variant="primary", scale=1)
+                    msg = gr.Textbox(
+                        label="",
+                        placeholder="Ask a complex research question...",
+                        scale=4,
+                        lines=1
+                    )
+                    submit_btn = gr.Button("Send", variant="primary", scale=1)
 
                 with gr.Row():
-                    clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
+                    clear_btn = gr.Button("Clear Chat", variant="secondary")
 
-                audio_output = gr.Audio(label="🔊 Voice Response", autoplay=True, interactive=False)
+                audio_output = gr.Audio(
+                    label="Voice Response",
+                    autoplay=True,
+                    interactive=False
+                )
 
             with gr.Column(scale=1):
                 with gr.Group():
-                    gr.HTML("<h3 style='text-align: center;'>📄 Upload Documents</h3>")
-                    file_upload = gr.Files(label="", file_types=[".pdf"], file_count="multiple")
-                upload_status = gr.Textbox(label="📊 Status", interactive=False, max_lines=10)
+                    gr.HTML("<h3 style='text-align: center;'>Upload Documents</h3>")
+                    file_upload = gr.Files(
+                        label="",
+                        file_types=[".pdf"],
+                        file_count="multiple"
+                    )
+                upload_status = gr.Textbox(
+                    label="Status",
+                    interactive=False,
+                    max_lines=10
+                )
 
         def respond(message, history):
+            """Handle user message"""
             new_hist, audio_file = agent.ask(message, history)
             return "", new_hist, audio_file
 
-        submit_btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot, audio_output])
-        msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot, audio_output])
-        clear_btn.click(lambda: ([], None), outputs=[chatbot, audio_output])
-        file_upload.change(agent.upload_pdfs, inputs=[file_upload], outputs=[upload_status])
+        def clear_chat():
+            """Clear chat history"""
+            return [], None
+
+        # Connect events
+        submit_btn.click(
+            respond,
+            inputs=[msg, chatbot],
+            outputs=[msg, chatbot, audio_output]
+        )
+        msg.submit(
+            respond,
+            inputs=[msg, chatbot],
+            outputs=[msg, chatbot, audio_output]
+        )
+        clear_btn.click(
+            clear_chat,
+            outputs=[chatbot, audio_output]
+        )
+        file_upload.change(
+            agent.upload_pdfs,
+            inputs=[file_upload],
+            outputs=[upload_status]
+        )
 
     return interface
 
+
 if __name__ == "__main__":
+    print("🚀 Starting AI Research Agent...")
     app = create_interface()
-    app.launch(server_name="0.0.0.0", server_port=7860)
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        share=False
+    )