Spaces:

ask-dev
/

HackBuddyAI

Sleeping

App Files Files Community

Mahdiyar commited on Jun 24

Commit

c47b1e1

1 Parent(s): 5e3d8c8

Improvement

Browse files

Files changed (2) hide show

app.py +59 -49
matching_agent.py +2 -1

app.py CHANGED Viewed

@@ -5,6 +5,7 @@ import logging
 import sys
 import os
 import io
 # Change from absolute imports to relative imports
 from database import initialize_database, add_participant, get_participants_dataframe
@@ -50,8 +51,16 @@ initialize_database()
 if not os.environ.get("OPENAI_API_KEY"):
     raise ValueError("The OPENAI_API_KEY environment variable is not set. Please set it before running the app.")
-logger.info("Creating matching agent...")
-agent = create_matching_agent(log_manager=log_manager)
 # --- Gradio UI Functions ---
@@ -84,14 +93,15 @@ def refresh_participants_list():
 async def run_matching_process(organizer_criteria, progress=gr.Progress(track_tqdm=True)):
     """Async callback to run the team matching process, with live log streaming."""
     # 1. Clear previous logs and show panel
     log_stream.seek(0)
     log_stream.truncate(0)
     progress(0, desc="Initializing...")
-    yield {
-        log_panel: gr.update(visible=True, open=True),
-        log_output: "Starting matching process...\n",
-        matching_results_out: "🚀 Setting up the matching process..."
-    }
     # 2. Get participants and perform initial checks
     logger.info("Fetching participants...")
@@ -100,47 +110,51 @@ async def run_matching_process(organizer_criteria, progress=gr.Progress(track_tq
     if len(participants_df) < 2:
         warning_msg = "Matching process aborted: not enough participants."
         logger.warning(warning_msg)
-        yield {
-            log_panel: gr.update(),
-            log_output: log_stream.getvalue(),
-            matching_results_out: "Cannot run matching with fewer than 2 participants."
-        }
         return
     logger.info(f"Running matching for {len(participants_df)} participants.")
     progress(0.2, desc="🧠 Agent is thinking...")
-    # 3. Create a background task for the core matching logic
-    match_task = asyncio.create_task(
-        run_matching(agent, participants_df, organizer_criteria)
-    )
-    # 4. Stream logs while the agent works
-    while not match_task.done():
-        await asyncio.sleep(0.5)  # Poll for new logs every 0.5s
-        yield {
-            log_panel: gr.update(),
-            log_output: log_stream.getvalue(),
-            matching_results_out: gr.update()  # No change to final output yet
-        }
-    # 5. Process the final result from the agent
-    try:
-        final_report = await match_task
-        logger.info("Matching process completed successfully.")
-        progress(1.0, desc="✅ Done!")
-        yield {
-            log_panel: gr.update(),
-            log_output: log_stream.getvalue(),
-            matching_results_out: final_report
-        }
-    except Exception as e:
-        logger.error(f"An error occurred during the matching process: {e}", exc_info=True)
-        yield {
-            log_panel: gr.update(),
-            log_output: log_stream.getvalue(),
-            matching_results_out: f"An error occurred: {e}"
-        }
 # --- Gradio App Definition ---
@@ -216,7 +230,7 @@ with gr.Blocks(theme=gr.themes.Default(
     run_button.click(
         fn=run_matching_process,
         inputs=[organizer_criteria_in],
-        outputs=[matching_results_out, log_output, log_panel]
     )
 # --- Launching the App ---
@@ -228,8 +242,4 @@ if __name__ == "__main__":
     except KeyboardInterrupt:
         logger.info("Gradio app shutting down.")
     finally:
-        # Clean up agent resources
-        logger.info("Closing agent resources...")
-        # We need to run the async close method
-        asyncio.run(agent.close())
-        logger.info("Cleanup complete.")

 import sys
 import os
 import io
+from contextlib import asynccontextmanager
 # Change from absolute imports to relative imports
 from database import initialize_database, add_participant, get_participants_dataframe
 if not os.environ.get("OPENAI_API_KEY"):
     raise ValueError("The OPENAI_API_KEY environment variable is not set. Please set it before running the app.")
+# --- Session Management ---
+@asynccontextmanager
+async def agent_session():
+    """Create and manage a session-specific agent."""
+    session_agent = create_matching_agent(log_manager=log_manager)
+    try:
+        yield session_agent
+    finally:
+        await session_agent.close()
+        logger.info("Session agent resources cleaned up")
 # --- Gradio UI Functions ---
 async def run_matching_process(organizer_criteria, progress=gr.Progress(track_tqdm=True)):
     """Async callback to run the team matching process, with live log streaming."""
     # 1. Clear previous logs and show panel
     log_stream.seek(0)
     log_stream.truncate(0)
     progress(0, desc="Initializing...")
+    yield [
+        gr.update(visible=True, open=True),  # log_panel
+        "Starting matching process...\n",     # log_output
+        "🚀 Setting up the matching process..."  # matching_results_out
+    ]
     # 2. Get participants and perform initial checks
     logger.info("Fetching participants...")
     if len(participants_df) < 2:
         warning_msg = "Matching process aborted: not enough participants."
         logger.warning(warning_msg)
+        yield [
+            gr.update(),  # log_panel
+            log_stream.getvalue(),  # log_output
+            "Cannot run matching with fewer than 2 participants."  # matching_results_out
+        ]
         return
     logger.info(f"Running matching for {len(participants_df)} participants.")
     progress(0.2, desc="🧠 Agent is thinking...")
+    # 3. Create a session-specific agent and run matching
+    async with agent_session() as session_agent:
+        logger.info("Created session-specific agent")
+        # Create a background task for the core matching logic
+        match_task = asyncio.create_task(
+            run_matching(session_agent, participants_df, organizer_criteria)
+        )
+        # 4. Stream logs while the agent works
+        while not match_task.done():
+            await asyncio.sleep(0.5)  # Poll for new logs every 0.5s
+            yield [
+                gr.update(),  # log_panel
+                log_stream.getvalue(),  # log_output
+                gr.update()  # matching_results_out - No change to final output yet
+            ]
+        # 5. Process the final result from the agent
+        try:
+            final_report = await match_task
+            logger.info("Matching process completed successfully.")
+            progress(1.0, desc="✅ Done!")
+            yield [
+                gr.update(),  # log_panel
+                log_stream.getvalue(),  # log_output
+                final_report  # matching_results_out
+            ]
+        except Exception as e:
+            logger.error(f"An error occurred during the matching process: {e}", exc_info=True)
+            yield [
+                gr.update(),  # log_panel
+                log_stream.getvalue(),  # log_output
+                f"An error occurred: {e}"  # matching_results_out
+            ]
 # --- Gradio App Definition ---
     run_button.click(
         fn=run_matching_process,
         inputs=[organizer_criteria_in],
+        outputs=[log_panel, log_output, matching_results_out]
     )
 # --- Launching the App ---
     except KeyboardInterrupt:
         logger.info("Gradio app shutting down.")
     finally:
+        logger.info("Shutdown complete.")

matching_agent.py CHANGED Viewed

@@ -49,7 +49,7 @@ def create_matching_agent(log_manager=None) -> TinyCodeAgent:
         api_key=os.environ.get("OPENAI_API_KEY"),
         log_manager=log_manager,
         pip_packages=["pandas", "numpy", "scikit-learn"],
-        #authorized_imports=["pandas", "numpy", "io", "base64","collections","itertools"],
         local_execution=False, # Use remote Modal for security by default
     )
@@ -94,6 +94,7 @@ async def run_matching(
     task = organizer_prompt+'\n\n'
     task += ("Form the teams based on the provided data and criteria."
                    f"\n<Organizer Criteria>\n{organizer_criteria}\n</Organizer Criteria>")
     final_report = await agent.run(task, max_turns=15)

         api_key=os.environ.get("OPENAI_API_KEY"),
         log_manager=log_manager,
         pip_packages=["pandas", "numpy", "scikit-learn"],
+        authorized_imports=["pandas", "numpy", "collections","itertools","requests"],
         local_execution=False, # Use remote Modal for security by default
     )
     task = organizer_prompt+'\n\n'
     task += ("Form the teams based on the provided data and criteria."
+             "\n Please go through all of them and give me details of all groups. "
                    f"\n<Organizer Criteria>\n{organizer_criteria}\n</Organizer Criteria>")
     final_report = await agent.run(task, max_turns=15)