resolverkatla commited on
Commit
b409e48
·
1 Parent(s): 6d3b2b6
Files changed (2) hide show
  1. app.py +20 -20
  2. requirements.txt +2 -3
app.py CHANGED
@@ -1,27 +1,27 @@
 
1
  import gradio as gr
2
- import google.generativeai as genai
3
  import os
4
 
5
- # Set up Gemini API key
6
- api_key = os.getenv("GEMINI_API_KEY")
 
 
 
 
 
 
 
 
 
 
7
 
8
- def chat_with_ai(prompt):
9
- """Simple chatbot using Google's Gemini AI."""
10
- if not api_key:
11
- return "Error: API key missing. Set GEMINI_API_KEY in Hugging Face secrets."
12
- try:
13
- genai.configure(api_key=api_key)
14
- available_models = [m.name for m in genai.list_models()]
15
- model_name = "gemini-pro" if "gemini-pro" in available_models else available_models[0]
16
- model = genai.GenerativeModel(model_name)
17
- response = model.generate_content(prompt)
18
- return response.text
19
- except Exception as e:
20
- return f"Error: {str(e)}"
21
 
22
- # Create Gradio chatbot interface
23
- iface = gr.ChatInterface(fn=chat_with_ai)
24
 
25
- # Launch the app
26
  if __name__ == "__main__":
27
- iface.launch()
 
1
+ import openai
2
  import gradio as gr
 
3
  import os
4
 
5
+ def chat_with_openai(prompt, history=[]):
6
+ messages = [{"role": "system", "content": "You are a helpful assistant."}]
7
+ for user_msg, bot_msg in history:
8
+ messages.append({"role": "user", "content": user_msg})
9
+ messages.append({"role": "assistant", "content": bot_msg})
10
+ messages.append({"role": "user", "content": prompt})
11
+
12
+ response = openai.ChatCompletion.create(
13
+ model="gpt-4", # Change to "gpt-4" if needed
14
+ messages=messages
15
+ )
16
+ return response["choices"][0]["message"]["content"]
17
 
18
+ def chatbot_interface(user_input, chat_history):
19
+ response = chat_with_openai(user_input, chat_history)
20
+ chat_history.append((user_input, response))
21
+ return response, chat_history
 
 
 
 
 
 
 
 
 
22
 
23
+ # Create a Gradio interface
24
+ demo = gr.ChatInterface(fn=chatbot_interface)
25
 
 
26
  if __name__ == "__main__":
27
+ demo.launch()
requirements.txt CHANGED
@@ -1,3 +1,2 @@
1
- google-generativeai
2
- gradio
3
- python-dotenv
 
1
+ openai
2
+ gradio