logging

app.py

@@ -1,12 +1,26 @@
 import gradio as gr
 import asyncio
+import logging
 from utils.generator import generate, generate_streaming
 
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.StreamHandler(),
+        logging.FileHandler('app.log')
+    ]
+)
+logger = logging.getLogger(__name__)
+
 # ---------------------------------------------------------------------
 # Wrapper function to handle async streaming for Gradio
 # ---------------------------------------------------------------------
 def generate_streaming_wrapper(query: str, context: str):
     """Wrapper to convert async generator to sync generator for Gradio"""
+    logger.info(f"Starting generation request - Query length: {len(query)}, Context length: {len(context)}")
+    
     async def _async_generator():
         async for chunk in generate_streaming(query, context):
             yield chunk
@@ -14,27 +28,33 @@ def generate_streaming_wrapper(query: str, context: str):
     # Create a new event loop for this thread
     try:
         loop = asyncio.get_event_loop()
+        logger.debug("Using existing event loop")
     except RuntimeError:
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
+        logger.debug("Created new event loop")
     
     # Convert async generator to sync generator
     async_gen = _async_generator()
     
     # Accumulate chunks for Gradio streaming
     accumulated_text = ""
+    chunk_count = 0
     
     while True:
         try:
             chunk = loop.run_until_complete(async_gen.__anext__())
             accumulated_text += chunk
+            chunk_count += 1
             yield accumulated_text  # Yield the accumulated text, not just the chunk
         except StopAsyncIteration:
+            logger.info(f"Generation completed - Total chunks: {chunk_count}, Final text length: {len(accumulated_text)}")
             break
 
 # ---------------------------------------------------------------------
 # Gradio Interface with MCP support and streaming
 # ---------------------------------------------------------------------
+logger.info("Initializing Gradio interface")
 ui = gr.Interface(
     fn=generate_streaming_wrapper,  # Use streaming wrapper function
     inputs=[
@@ -63,6 +83,8 @@ ui = gr.Interface(
 
 # Launch with MCP server enabled
 if __name__ == "__main__":
+    logger.info("Starting ChatFed Generation Module server")
+    logger.info("Server will be available at http://0.0.0.0:7860")
     ui.launch(
         server_name="0.0.0.0",
         server_port=7860,