Update app.py

app.py

@@ -1,34 +1,153 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
+import gradio as gr
 from datetime import datetime
 import pytz
+import json
 
 # Pakistan timezone
 pakistan_tz = pytz.timezone("Asia/Karachi")
 
-# FastAPI app
-app = FastAPI(title="Pakistan Time Bot API")
+def chatbot(message):
+    # Handle JSON input from ESP8266
+    try:
+        # Parse the JSON input
+        data = json.loads(message)
+        
+        # Extract the actual message from the "data" array
+        if "data" in data and isinstance(data["data"], list) and len(data["data"]) > 0:
+            user_message = data["data"][0].lower()
+        else:
+            return json.dumps({"data": ["I can only tell you the current time in Pakistan. Try asking: 'What is the time?'"]})
+    except:
+        # If it's not JSON, treat it as a regular string (for direct Gradio interface testing)
+        user_message = message.lower()
+    
+    # Process the message
+    if "time" in user_message:
+        now_pk = datetime.now(pakistan_tz)
+        time_str = f"{now_pk.strftime('%I:%M %p')}"
+        # Return in the expected JSON format
+        return json.dumps({"data": [time_str]})
+    else:
+        return json.dumps({"data": ["I can only tell you the current time in Pakistan. Try asking: 'What is the time?'"]})
 
-# Schema for ESP8266 request
-class RequestData(BaseModel):
-    data: list[str]
+# Create a more comprehensive interface
+with gr.Blocks(title="Pakistan Time Bot", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🇵🇰 Pakistan Time Bot")
+    gr.Markdown("This bot tells you the current time in Pakistan. It's designed to work with both direct queries and ESP8266 devices.")
+    
+    with gr.Tab("Chat Interface"):
+        gr.Markdown("### Talk to the Time Bot")
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.Textbox(label="Your Message", placeholder="Ask me about the time in Pakistan...")
+                text_button = gr.Button("Get Time")
+            with gr.Column():
+                output_text = gr.Textbox(label="Bot Response", interactive=False)
+        
+        examples = gr.Examples(
+            examples=["What's the time?", "Time please", "Current time in Pakistan"],
+            inputs=input_text
+        )
+    
+    with gr.Tab("ESP8266 Integration"):
+        gr.Markdown("### ESP8266 Integration")
+        gr.Markdown("""
+        This bot is designed to work with ESP8266 microcontrollers. Here's how to set it up:
+        
+        1. Use the code below in your Arduino IDE
+        2. Replace `YOUR_WIFI_SSID` and `YOUR_WIFI_PASSWORD` with your WiFi credentials
+        3. Update the `serverName` to point to your deployed Hugging Face Space
+        4. Upload the code to your ESP8266
+        """)
+        
+        esp_code = """
+#include <ESP8266WiFi.h>
+#include <ESP8266HTTPClient.h>
 
-@app.get("/")
-def home():
-    return {"message": "🇵🇰 Pakistan Time Bot API is running!"}
+const char* ssid = "YOUR_WIFI_SSID";
+const char* password = "YOUR_WIFI_PASSWORD";
 
-@app.post("/run/predict")
-async def get_time(request: RequestData):
-    # Validate and extract message
-    if request.data and isinstance(request.data, list):
-        user_message = request.data[0].lower()
-    else:
-        return {"data": ["Invalid input. Use: {\"data\": [\"time\"]}"]}
+// Hugging Face Space API endpoint
+String serverName = "https://your-username-time-bot.hf.space/run/predict";
 
-    # Handle time request
-    if "time" in user_message:
-        now_pk = datetime.now(pakistan_tz)
-        time_str = now_pk.strftime("%I:%M %p")
-        return {"data": [time_str]}
-    else:
-        return {"data": ["I can only tell you the current time in Pakistan. Try asking: 'What is the time?'"]}
+WiFiClientSecure client; // use WiFiClientSecure for HTTPS
+
+void setup() {
+  Serial.begin(115200);
+  WiFi.begin(ssid, password);
+
+  Serial.print("Connecting to WiFi...");
+  while (WiFi.status() != WL_CONNECTED) {
+    delay(500);
+    Serial.print(".");
+  }
+  Serial.println(" Connected!");
+
+  // For HTTPS: skip certificate verification
+  client.setInsecure();
+}
+
+void loop() {
+  if (WiFi.status() == WL_CONNECTED) {
+    HTTPClient http;
+
+    http.begin(client, serverName);
+    http.addHeader("Content-Type", "application/json");
+
+    // Sending request
+    String jsonData = "{\\"data\\":[\\"time\\"]}";
+    int httpResponseCode = http.POST(jsonData);
+
+    if (httpResponseCode > 0) {
+      String response = http.getString();
+      Serial.println("Response:");
+      Serial.println(response);
+    } else {
+      Serial.print("Error code: ");
+      Serial.println(httpResponseCode);
+    }
+
+    http.end();
+  } else {
+    Serial.println("WiFi Disconnected");
+  }
+
+  delay(10000); // request every 10 seconds
+}
+        """
+        
+        gr.Code(value=esp_code, language="cpp", label="ESP8266 Code")
+        
+        gr.Markdown("### Test ESP8266 Request")
+        gr.Markdown("Click the button below to simulate the JSON request that an ESP8266 would send:")
+        test_button = gr.Button("Test ESP8266 Request")
+        test_output = gr.Textbox(label="JSON Response", interactive=False)
+    
+    with gr.Tab("API Info"):
+        gr.Markdown("### API Information")
+        gr.Markdown("""
+        The bot accepts two types of requests:
+        
+        1. **Direct text input**: Simply send a text message containing the word "time"
+        2. **JSON API** (for ESP8266): Send a JSON object in the format `{"data": ["time"]}`
+        
+        The response will always be a JSON object with the format:
+        ```json
+        {"data": ["08:45 PM"]}
+        ```
+        """)
+        
+        gr.Markdown("### Direct API Testing")
+        with gr.Row():
+            api_input = gr.Textbox(value='{"data": ["time"]}', label="JSON Input")
+            api_button = gr.Button("Send API Request")
+        api_output = gr.Textbox(label="API Response")
+    
+    # Set up event handlers
+    text_button.click(chatbot, inputs=input_text, outputs=output_text)
+    input_text.submit(chatbot, inputs=input_text, outputs=output_text)
+    test_button.click(chatbot, inputs=gr.Textbox(value='{"data": ["time"]}'), outputs=test_output)
+    api_button.click(chatbot, inputs=api_input, outputs=api_output)
+
+if __name__ == "__main__":
+    demo.launch()