Update app.py

app.py

@@ -6,6 +6,7 @@ import base64
 import io
 import json
 import re
+import logging
 from datetime import datetime, timedelta
 
 # Third-party imports
@@ -26,6 +27,9 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 
+# Setup logging
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+
 # Load environment variables
 load_dotenv()
 client = groq.Client(api_key=os.getenv("GROQ_LEGAL_API_KEY"))
@@ -78,7 +82,7 @@ class ModelName(str):
             raise ValueError(f"Invalid model name: {v}")
         return v
 
-# Custom CSS with dark mode support
+# Custom CSS
 custom_css = """
 :root {
     --bg-color: #FFFFFF;
@@ -125,7 +129,7 @@ body { background-color: var(--bg-color); color: var(--text-color); font-family:
 @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
 """
 
-# Custom JavaScript for UI enhancements
+# Custom JavaScript (simplified to avoid interference)
 custom_js = """
 function toggleTheme() {
     const currentTheme = document.body.getAttribute('data-theme');
@@ -147,22 +151,19 @@ document.addEventListener('DOMContentLoaded', () => {
     if (savedTheme) {
         document.body.setAttribute('data-theme', savedTheme);
     }
-    tippy('#pdf_file', { content: 'Upload a PDF document for analysis', placement: 'top' });
-    tippy('#ticker_input', { content: 'Enter a stock ticker symbol (e.g., AAPL)', placement: 'top' });
 });
 """
 
-# Spinner HTML and Tippy.js for tooltips
+# Spinner HTML
 custom_html = """
 <div id="spinner" style="display: none; position: fixed; top: 50%; left: 50%; transform: translate(-50%, -50%);">
     <div class="spinner"></div>
 </div>
-<script src="https://unpkg.com/@popperjs/core@2"></script>
-<script src="https://unpkg.com/tippy.js@6"></script>
 """
 
 # Helper Functions
 def process_pdf(pdf_file):
+    logging.debug("Processing PDF file")
     if pdf_file is None:
         return None, "No file uploaded", PDFState(page_images=[], total_pages=0, total_words=0)
     
@@ -188,8 +189,10 @@ def process_pdf(pdf_file):
 
         os.unlink(pdf_path)
         pdf_state = PDFState(page_images=page_images, total_pages=total_pages, total_words=total_words)
+        logging.debug(f"PDF processed successfully: {session_id}")
         return session_id, f"✅ Successfully processed {len(chunks)} text chunks from your PDF", pdf_state
     except Exception as e:
+        logging.error(f"Error processing PDF: {str(e)}")
         if "pdf_path" in locals() and os.path.exists(pdf_path):
             os.unlink(pdf_path)
         return None, f"Error processing PDF: {str(e)}", PDFState(page_images=[], total_pages=0, total_words=0)
@@ -308,9 +311,11 @@ def create_stock_chart(ticker, period, enable_stock_data):
     return fig
 
 def analyze_ticker(ticker_input, period, use_brave_search, enable_stock_data, enable_search):
+    logging.debug(f"Analyzing ticker: {ticker_input}, period: {period}")
     try:
         input_data = StockAnalysisInput(ticker=ticker_input, period=period)
     except ValueError as e:
+        logging.error(f"Validation error: {str(e)}")
         return None, str(e), None
     
     ticker = input_data.ticker
@@ -333,17 +338,21 @@ def analyze_ticker(ticker_input, period, use_brave_search, enable_stock_data, en
 **P/E Ratio:** {stock_data['pe_ratio']}  
 **Market Sentiment:** {sentiment}
 """
+        logging.debug(f"Ticker {ticker} analyzed successfully")
         return chart, summary, ticker
     except Exception as e:
+        logging.error(f"Error analyzing ticker {ticker}: {str(e)}")
         return None, f"Error analyzing ticker {ticker}: {str(e)}", None
 
 def generate_response(message, session_id, model_name, history, current_ticker, use_brave_search, enable_search, enable_stock_data):
+    logging.debug(f"Generating response for message: {message}")
     if not message:
         return history
     
     try:
         model_name = ModelName.validate(model_name)
     except ValueError as e:
+        logging.error(f"Model validation error: {str(e)}")
         return history + [(message, str(e))]
     
     try:
@@ -368,6 +377,7 @@ def generate_response(message, session_id, model_name, history, current_ticker,
             for i, item in enumerate(news[:3]):
                 response += f"{i+1}. [{item['title']}]({item['link']})\n   {item['snippet'][:100]}...\n"
             history.append((message, response))
+            logging.debug(f"Stock response generated for {ticker}")
             return history
         
         if message.lower().startswith("/news "):
@@ -377,6 +387,7 @@ def generate_response(message, session_id, model_name, history, current_ticker,
             for i, item in enumerate(news[:5]):
                 response += f"{i+1}. **{item['title']}**\n   {item['snippet']}\n   [Read more]({item['link']})\n\n"
             history.append((message, response))
+            logging.debug(f"News response generated for {topic}")
             return history
         
         system_prompt = "You are a financial assistant specializing in analyzing financial reports, statements, and market trends."
@@ -390,10 +401,11 @@ def generate_response(message, session_id, model_name, history, current_ticker,
         )
         response = completion.choices[0].message.content
         history.append((message, response))
+        logging.debug("Chat response generated")
         return history
     except Exception as e:
-        history.append((message, f"Error generating response: {str(e)}"))
-        return history
+        logging.error(f"Error generating response: {str(e)}")
+        return history + [(message, f"Error generating response: {str(e)}")]
 
 def update_pdf_viewer(pdf_state: PDFState):
     if not pdf_state.total_pages:
@@ -466,48 +478,73 @@ def create_interface():
         
         # Event Handlers
         upload_button.click(
-            js="showSpinner",
             fn=process_pdf,
             inputs=[pdf_file],
-            outputs=[current_session_id, pdf_status, pdf_state]
+            outputs=[current_session_id, pdf_status, pdf_state],
+            _js="showSpinner"
         ).then(
-            update_pdf_viewer,
+            fn=update_pdf_viewer,
             inputs=[pdf_state],
             outputs=[page_slider, pdf_image, stats_display]
         ).then(
-            js="hideSpinner",
+            fn=None,
+            _js="hideSpinner",
             inputs=[],
             outputs=[]
         )
         
         analyze_button.click(
-            js="showSpinner",
             fn=analyze_ticker,
             inputs=[ticker_input, period_dropdown, use_brave_search, enable_stock_data, enable_search],
-            outputs=[stock_chart, stock_summary, current_ticker]
+            outputs=[stock_chart, stock_summary, current_ticker],
+            _js="showSpinner"
         ).then(
-            js="hideSpinner",
+            fn=None,
+            _js="hideSpinner",
             inputs=[],
             outputs=[]
         )
         
         msg.submit(
-            generate_response,
+            fn=generate_response,
             inputs=[msg, current_session_id, model_dropdown, chatbot, current_ticker, use_brave_search, enable_search, enable_stock_data],
             outputs=[chatbot]
-        ).then(lambda: "", None, [msg])
+        ).then(
+            fn=lambda: "",
+            inputs=None,
+            outputs=[msg]
+        )
         
         send_btn.click(
-            generate_response,
+            fn=generate_response,
             inputs=[msg, current_session_id, model_dropdown, chatbot, current_ticker, use_brave_search, enable_search, enable_stock_data],
-            outputs=[chatbot]
-        ).then(lambda: "", None, [msg])
+            outputs=[chatbot],
+            _js="showSpinner"
+        ).then(
+            fn=lambda: "",
+            inputs=None,
+            outputs=[msg]
+        ).then(
+            fn=None,
+            _js="hideSpinner",
+            inputs=[],
+            outputs=[]
+        )
         
-        page_slider.change(update_image, inputs=[page_slider, pdf_state], outputs=[pdf_image])
-        theme_button.click(js="toggleTheme")
+        page_slider.change(
+            fn=update_image,
+            inputs=[page_slider, pdf_state],
+            outputs=[pdf_image]
+        )
+        theme_button.click(
+            fn=None,
+            _js="toggleTheme",
+            inputs=[],
+            outputs=[]
+        )
     
     return demo
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch(debug=True)  # Enable debug mode to see logs in console