attempting to ad loging

app.py

@@ -50,10 +50,12 @@ dataset = load_dataset('tosin2013/autogen', streaming=True)
 dataset = Dataset.from_list(list(dataset['train']))
 
 # Initialize embeddings
+print("[EMBEDDINGS] Loading sentence-transformers model...")
 embeddings = HuggingFaceEmbeddings(
     model_name="sentence-transformers/all-MiniLM-L6-v2",
     model_kwargs={"device": "cpu"}
 )
+print("[EMBEDDINGS] Sentence-transformers model loaded successfully")
 
 # Extract texts from the dataset
 texts = dataset['input']
@@ -61,7 +63,9 @@ texts = dataset['input']
 # Create and cache embeddings for the texts
 if not os.path.exists('embeddings.npy'):
     print("[LOG] Generating embeddings...")
+    print("[EMBEDDINGS] Generating document embeddings...")
     text_embeddings = embeddings.embed_documents(texts)
+    print(f"[EMBEDDINGS] Generated embeddings for {len(texts)} documents")
     np.save('embeddings.npy', text_embeddings)
 else:
     print("[LOG] Loading cached embeddings...")
@@ -88,7 +92,9 @@ def get_relevant_documents(query, k=5):
     import time
     start_time = time.time()
     
+    print("[EMBEDDINGS] Generating embedding for query...")
     query_embedding = embeddings.embed_query(query)
+    print("[EMBEDDINGS] Query embedding generated successfully")
     distances, indices = nn.kneighbors([query_embedding], n_neighbors=k)
     relevant_docs = [texts[i] for i in indices[0]]
     
@@ -290,6 +296,7 @@ Provide the AutoGen v0.4 agent code that fulfills the user's request. Utilize fe
 
 
 # Create Gradio interface
+print("[CHAT] Initializing chat interface...")
 with gr.Blocks() as demo:
     gr.Markdown(f"""
     ## AutoGen v0.4 Agent Code Generator QA Agent
@@ -327,12 +334,14 @@ with gr.Blocks() as demo:
         outputs=[chatbot],
         queue=True
     )
+    print("[CHAT] Submit button handler configured")
     
     clear_btn.click(
         lambda: (None, ""),
         inputs=[],
         outputs=[chatbot, question]
     )
+    print("[CHAT] Clear button handler configured")
 
 import socket
 
@@ -350,14 +359,7 @@ if __name__ == "__main__":
     try:
         port = find_available_port()
         print(f"[LOG] Launching application on port {port}")
+        print("[CHAT] Starting chat server...")
         demo.launch()
-        # Verify server is actually running
-        import time
-        time.sleep(2)  # Give server time to start
-        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-            if s.connect_ex(('localhost', port)) == 0:
-                print(f"[SUCCESS] Server is running on port {port}")
-            else:
-                print(f"[ERROR] Failed to bind to port {port}")
     except Exception as e:
         print(f"[ERROR] Failed to start application: {str(e)}")