Update app.py

app.py

@@ -1,4 +1,3 @@
-import gradio as gr
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse
 import datetime
@@ -92,54 +91,4 @@ async def chat_completion(request: Request):
         raise e
     except Exception as e:
         logger.error(f"Unexpected error: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
-
-def generate_response(messages):
-    payload = {
-        "inputs": {
-            "messages": messages
-        },
-        "parameters": {
-            "max_new_tokens": 2048,
-            "temperature": 0.7,
-            "top_p": 0.95,
-            "do_sample": True
-        }
-    }
-    
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        result = response.json()
-        
-        if isinstance(result, dict) and "error" in result:
-            return f"Error: {result['error']}"
-        
-        return result[0]["generated_text"]
-    except requests.exceptions.RequestException as e:
-        logger.error(f"Request failed: {e}")
-        return f"Error: {e}"
-
-def chat_interface(messages):
-    chat_history = []
-    for message in messages:
-        try:
-            response = generate_response([{"role": "user", "content": message}])
-            chat_history.append({"role": "user", "content": message})
-            chat_history.append({"role": "assistant", "content": response})
-        except Exception as e:
-            chat_history.append({"role": "user", "content": message})
-            chat_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
-    return chat_history
-
-# Create Gradio interface
-def gradio_app():
-    return gr.ChatInterface(chat_interface, type="messages")
-
-# Mount both FastAPI and Gradio
-app = gr.mount_gradio_app(app, gradio_app(), path="/")
-
-# For running with uvicorn directly
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+        raise HTTPException(status_code=500, detail=str(e))