Update app.py

app.py

@@ -1,153 +1,34 @@
-import gradio as gr
+from fastapi import FastAPI
+from pydantic import BaseModel
 from datetime import datetime
 import pytz
-import json
 
 # Pakistan timezone
 pakistan_tz = pytz.timezone("Asia/Karachi")
 
-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?'"]})
-
-# 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>
-
-const char* ssid = "YOUR_WIFI_SSID";
-const char* password = "YOUR_WIFI_PASSWORD";
-
-// Hugging Face Space API endpoint
-String serverName = "https://your-username-time-bot.hf.space/run/predict";
-
-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!");
+# FastAPI app
+app = FastAPI(title="Pakistan Time Bot API")
 
-  // For HTTPS: skip certificate verification
-  client.setInsecure();
-}
+# Schema for ESP8266 request
+class RequestData(BaseModel):
+    data: list[str]
 
-void loop() {
-  if (WiFi.status() == WL_CONNECTED) {
-    HTTPClient http;
+@app.get("/")
+def home():
+    return {"message": "🇵🇰 Pakistan Time Bot API is running!"}
 
-    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)
+@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\"]}"]}
 
-if __name__ == "__main__":
-    demo.launch()
+    # 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?'"]}