Upload 2 files

app.py

@@ -0,0 +1,145 @@
+import os
+import json
+import requests
+from openai import OpenAI
+import gradio as gr
+
+# --- Environment Variables (set these in your Hugging Face Space settings) ---
+# Add under "Settings → Variables":
+#   OPENAI_API_KEY = your_openai_key
+#   STOCK_API_KEY = your_alphavantage_key
+
+os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
+os.environ["STOCK_API_KEY"] = os.getenv("STOCK_API_KEY")
+
+client = OpenAI()
+
+# --- Tool Function: Fetch latest stock price ---
+def get_latest_stock_price(symbol: str) -> str:
+    api_key = os.environ.get("STOCK_API_KEY")
+    base_url = "https://www.alphavantage.co/query"
+    params = {
+        "function": "TIME_SERIES_DAILY",
+        "symbol": symbol,
+        "apikey": api_key,
+    }
+    response = requests.get(base_url, params=params)
+    data = response.json()
+
+    if "Time Series (Daily)" in data and data["Time Series (Daily)"]:
+        latest_day_data = list(data["Time Series (Daily)"].values())[0]
+        if "4. close" in latest_day_data:
+            latest_price = latest_day_data["4. close"]
+            return latest_price
+        else:
+            return "Price data not available."
+    else:
+        return "Price data not available."
+
+# --- Define Tools for OpenAI Function Calling ---
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "get_latest_stock_price",
+            "description": "Get the latest stock price for a given symbol",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "symbol": {
+                        "type": "string",
+                        "description": "The symbol of the stock",
+                    },
+                },
+                "required": ["symbol"],
+            },
+        },
+    }
+]
+
+# --- Chat Logic ---
+def stock_chat(user_message: str) -> str:
+    messages = [
+        {
+            "role": "user",
+            "content": "You are a stock bot. Answer only finance stock related questions.",
+        },
+        {"role": "user", "content": user_message},
+    ]
+
+    response = client.chat.completions.create(
+        model="gpt-3.5-turbo-1106",
+        messages=messages,
+        tools=tools,
+        tool_choice="auto",
+    )
+
+    msg = response.choices[0].message
+
+    # --- If model calls tool ---
+    if hasattr(msg, "tool_calls") and msg.tool_calls:
+        for call in msg.tool_calls:
+            if call.function.name == "get_latest_stock_price":
+                try:
+                    args = json.loads(call.function.arguments or "{}")
+                except json.JSONDecodeError:
+                    args = {}
+
+                if "symbol" in args:
+                    symbol = args["symbol"]
+                    price = get_latest_stock_price(symbol)
+
+                    # Add assistant tool call + tool result
+                    messages.append(
+                        {
+                            "role": "assistant",
+                            "tool_calls": [
+                                {
+                                    "id": call.id,
+                                    "type": "function",
+                                    "function": {
+                                        "name": "get_latest_stock_price",
+                                        "arguments": json.dumps({"symbol": symbol}),
+                                    },
+                                }
+                            ],
+                            "content": None,
+                        }
+                    )
+
+                    messages.append(
+                        {
+                            "role": "tool",
+                            "tool_call_id": call.id,
+                            "content": f"The latest stock price for {symbol} is {price}",
+                        }
+                    )
+
+                    # Model crafts the final user-facing answer
+                    final = client.chat.completions.create(
+                        model="gpt-3.5-turbo-1106",
+                        messages=messages,
+                    )
+                    return final.choices[0].message.content or "No reply"
+                else:
+                    return "Symbol not found in tool call arguments."
+            else:
+                return "Unknown tool called."
+    else:
+        # Model answered directly — no tool required
+        return msg.content or "No reply"
+
+# --- Gradio UI for Hugging Face ---
+def chatbot_interface(message):
+    return stock_chat(message)
+
+iface = gr.Interface(
+    fn=chatbot_interface,
+    inputs=gr.Textbox(label="Ask about a stock (e.g., 'What is the latest stock price for AAPL?')"),
+    outputs=gr.Textbox(label="Response"),
+    title="📈 StockBot - AI Stock Assistant",
+    description="Ask questions about stock prices. Data provided by AlphaVantage and OpenAI GPT-3.5.",
+)
+
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+openai
+requests
+gradio