Update app.py

app.py

@@ -1,35 +1,24 @@
-import uvicorn
 import logging
 import os
 from fastapi import FastAPI, Request
-import httpx
+from contextlib import asynccontextmanager
 from transformers import pipeline
 from langdetect import detect
 from huggingface_hub import login
-
-
-app = FastAPI()
+import socket
 
 # Global variables
 HF_HUB_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
-
 # Verify Hugging Face token
 if not HF_HUB_TOKEN:
      raise ValueError("Missing Hugging Face API token. Please set HUGGINGFACEHUB_API_TOKEN in environment variables.")
 login(token=HF_HUB_TOKEN)
 
-
-# Configure logging
-logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-    
 # Load Hebrew and English text generation models
 hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
 english_generator = pipeline("text-generation", model="distilgpt2")
 
-
 # Function to detect language
 def detect_language(user_input):
     try:
@@ -38,7 +27,6 @@ def detect_language(user_input):
     except:
         return "unsupported"
 
-
 # Function to generate a response
 def generate_response(text):
     language = detect_language(text)
@@ -48,11 +36,31 @@ def generate_response(text):
         return english_generator(text, max_length=100)[0]["generated_text"]
     return "Sorry, I only support Hebrew and English."
 
+# FastAPI lifespan event
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("Starting application...")
+    yield  # Wait until app closes
+    print("Shutting down application...")
+
+# Create FastAPI app
+app = FastAPI(lifespan=lifespan)
 
 @app.get("/")
 async def root():
-    return {"message": "Server is running on HF Spaces"}
-
+    return {"message": "Decision Helper API is running!"}
+
+@app.post("/generate_response")
+async def generate_text(request: Request):
+    data = await request.json()
+    text = data.get("text", "")
+    if not text:
+        return {"error": "No text provided"}
+    
+    response = generate_response(text)
+    return {"response": response}
 
+# Run the server
 if __name__ == "__main__":
-   ### uvicorn.run(app, host="0.0.0.0", port=7860)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)