Update app.py

app.py

@@ -5,11 +5,12 @@ from multi_agent import run_multi_agent
 
 lock = threading.Lock()
 
-LLM = "gpt-4o"
+LLM = "llama3-8b-8192" #gpt-4o"
 
-def invoke(openai_api_key, task):
-    if not openai_api_key:
-        raise gr.Error("OpenAI API Key is required.")
+def invoke(groq_api_key, task):
+    if not GROQ_API_KEY_defualt:
+      if not groq_api_key:
+         raise gr.Error("Groq API Key is required.")
 
     if not task:
         raise gr.Error("Task is required.")
@@ -17,15 +18,15 @@ def invoke(openai_api_key, task):
     raise gr.Error("Please clone space due to local code execution.")
     
     with lock:
-        os.environ["OPENAI_API_KEY"] = openai_api_key
+        os.environ["GROQ_API_KEY"] = groq_api_key
         result = run_multi_agent(LLM, task)
-        del os.environ["OPENAI_API_KEY"]
+        del os.environ["GROQ_API_KEY"]
         return result
 
 gr.close_all()
 
 demo = gr.Interface(fn = invoke, 
-                    inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
+                    inputs = [gr.Textbox(label = "GROQ API KEY", type = "password", lines = 1),
                               gr.Textbox(label = "Task", value = os.environ['INPUT'])],
                     outputs = [gr.Markdown(label = "Output", value = os.environ["OUTPUT"], line_breaks = True, sanitize_html = False)],
                     title = "Multi-Agent AI: Coding",