Update app.py

app.py

@@ -1,15 +1,7 @@
 import logging
 from flask import Flask, render_template, request, jsonify
 import requests
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-from typing import List
-from gradio_client import Client
 import os
-import nest_asyncio
-
-nest_asyncio.apply()
-
 
 app = Flask(__name__)
 logging.basicConfig(level=logging.DEBUG)
@@ -27,8 +19,6 @@ class ChatContext:
         return "\n".join([f"{msg['role']}: {msg['content']}" for msg in self.messages])
 
 chat_context = ChatContext()
-# Initialize the client (this loads the API)
-client = Client(os.getenv("CLIENT_URL"))
 
 # System message
 system_message = """You are Echo 1.5, an AI assistant developed by Abhinav. Your purpose is to assist users with their queries and provide helpful information. You should always be polite, respectful, and informative in your responses.
@@ -45,26 +35,25 @@ Iamwomen: iamwomen.cloudns.be
 Iamwomen_Instagram: https://www.instagram.com/i_am_woman.24/
 """
 
-# Define the input model
-class ChatInput(BaseModel):
-    messages: List[dict]
-
 def chat_with_ai(context):
     # Prepare the full prompt
-    #full_prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in context])
-
+    full_prompt = context.get_context()
+    
     # Make the API call
-    result = client.predict(
-        message=full_prompt,
-        request=os.getenv('MODEL'),
-        param_3=0.5,
-        param_4=4096,
-        param_5=0.5,
-        param_6=0,
-        api_name="/chat"
-    )
+    try:
+        response = requests.post(API_URL, json={"prompt": full_prompt})
+        response.raise_for_status()
+        return response.json()["response"]
+    except requests.RequestException as e:
+        app.logger.error(f"Error communicating with API: {str(e)}")
+        raise Exception(f"Error communicating with API: {str(e)}")
+    except KeyError as e:
+        app.logger.error(f"Unexpected response format: {response.text}")
+        raise Exception(f"Unexpected response format: {str(e)}")
+    except Exception as e:
+        app.logger.error(f"Unexpected error: {str(e)}")
+        raise Exception(f"An unexpected error occurred: {str(e)}")
 
-    return result
 @app.route('/')
 def index():
     return render_template('index.html')
@@ -83,20 +72,13 @@ def chat():
     
     app.logger.debug(f"Full context: {full_context}")
     
+    chat_context.messages = conversation_history
+    chat_context.add_message('user', user_input)
+    
     try:
-            response = chat_with_ai(context=full_context)
-        )
-        app.logger.debug(f"API response status: {response.status_code}")
-        app.logger.debug(f"API response content: {response.text}")
-        response.raise_for_status()
-        ai_response = response.json()["response"]
-        return jsonify({"response": ai_response})
-    except requests.RequestException as e:
-        app.logger.error(f"Error communicating with API: {str(e)}")
-        return jsonify({"error": f"Error communicating with API: {str(e)}"}), 500
-    except KeyError as e:
-        app.logger.error(f"Unexpected response format: {response.text}")
-        return jsonify({"error": f"Unexpected response format: {str(e)}"}), 500
+        response = chat_with_ai(chat_context)
+        app.logger.debug(f"API response content: {response}")
+        return jsonify({"response": response})
     except Exception as e:
         app.logger.error(f"Unexpected error: {str(e)}")
         return jsonify({"error": f"An unexpected error occurred: {str(e)}"}), 500