Update app.py

update to use gpt

app.py

@@ -1,29 +1,38 @@
-from transformers import Conversation
-from huggingface_hub import login
+import gradio as gr
+import openai
 import os
-login(token = os.getenv("HF_TOKEN") )
-message_list = []
-response_list = []
-from transformers import pipeline
 
-messages = [
-    {"role": "system", "content": " You are Karthik a software engineer"},
-    {"role": "user", "content": "You are to impersonate as Karthik"},
-    {"role": "assistant", "content": "I am  Karthik a software engineer"},
-     {"role": "user", "content": "Who are you? "},
-]
-chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3",max_new_tokens=5000)
+# Load your OpenAI API key from the environment variable
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
+# Read the static CV file
+def load_cv():
+    with open("templated_CV.txt", 'r') as file:
+        return file.read()
 
-def mini_chatbot(message, history):
-    conversation = Conversation(text=message,
-                                past_user_inputs=message_list,
-                                generated_responses=response_list)
-    conversation = chatbot(conversation)
+# Extract information from the CV
+cv_text = load_cv()
 
-    return conversation.generated_responses[-1]
+def chat_with_ai(user_input):
+    prompt = f"{cv_text}\n\nUser: {user_input}\nAI:"
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message['content']
 
-demo_chatbot = gr.ChatInterface(mini_chatbot,
-                                title="My Chatbot",
-                                description="Enter text to start chatting.")
-demo_chatbot.launch()
+def main(user_input):
+    response = chat_with_ai(user_input)
+    return response
+
+iface = gr.Interface(
+    fn=main,
+    inputs=gr.Textbox(label="Ask a question, that you would like to ask Karthik"),
+    outputs="text",
+    title="AI Clone",
+    description="Interact with an AI clone for recruiting or for fun :) "
+)
+
+iface.launch()