Update app.py

app.py

@@ -1,40 +1,35 @@
-import gradio as gr
 import os
 
-from openai import OpenAI
+from crews import get_financial_trading_crew
 
-config = {
-    "max_tokens": 1000,
-    "model": "gpt-4",
-    "temperature": 0
-}
-
-def invoke(openai_api_key, prompt):
+def invoke(openai_api_key, serper_api_key, stock_selection, initial_capital, risk_tolerance, trading_strategy_preference, news_impact_consideration):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
-    if (prompt == ""):
-        raise gr.Error("Prompt is required.")
+    if (serp_api_key == ""):
+        raise gr.Error("SERPER API Key is required.")
+    if (stock_selection == ""):
+        raise gr.Error("Stock Ticker is required.")
+    if (initial_capital == ""):
+        raise gr.Error("Initial Capital is required.")
+    if (risk_tolerance == ""):
+        raise gr.Error("Risk Tolerance is required.")
+    if (trading_strategy_preference == ""):
+        raise gr.Error("Trading Strategy Preference is required.")
+    if (news_impact_consideration == ""):
+        raise gr.Error("News Impact Consideration is required.")
 
     os.environ["OPENAI_API_KEY"] = openai_api_key
+    os.environ["SERPER_API_KEY"] = serper_api_key
     
-    content = ""
-    
-    try:
-        client = OpenAI()
-    
-        completion = client.chat.completions.create(
-            max_tokens = config["max_tokens"],
-            messages = [{"role": "user", "content": prompt}],
-            model = config["model"],
-            temperature = config["temperature"])
+    financial_trading_inputs = {
+        'stock_selection': stock_selection,
+        'initial_capital': initial_capital,
+        'risk_tolerance': risk_tolerance,
+        'trading_strategy_preference': trading_strategy_preference,
+        'news_impact_consideration': news_impact_consideration
+    }
     
-        content = completion.choices[0].message.content
-    except Exception as e:
-        err_msg = e
-
-        raise gr.Error(e)
-
-    return content
+    return financial_trading_crew.kickoff(inputs=financial_trading_inputs)
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a href='https://openai.com/'>OpenAI</a> SDK 
                  with <a href='https://openai.com/research/gpt-4'>gpt-4</a> model."""
@@ -43,9 +38,14 @@ gr.close_all()
 
 demo = gr.Interface(fn = invoke, 
                     inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
-                              gr.Textbox(label = "Prompt", lines = 1)],
-                    outputs = [gr.Textbox(label = "Completion", lines = 1)],
-                    title = "Generative AI - LLM",
+                              gr.Textbox(label = "SERPER API Key", type = "password", lines = 1),
+                              gr.Textbox(label = "Stock Ticker", value="SAN", lines = 1),
+                              gr.Textbox(label = "Initial Capital", value=1000, lines = 1),
+                              gr.Textbox(label = "Risk Tolerance", value="Medium", lines = 1),
+                              gr.Textbox(label = "Trading Strategy Preference", value="Day Trading", lines = 1),
+                              gr.Textbox(label = "News Impact Consideration", value=True, lines = 1)],
+                    outputs = [gr.Textbox(label = "Output", lines = 1)],
+                    title = "Agentic RAG: Financial Analysis",
                     description = description)
 
 demo.launch()