Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import requests # For HTTP requests to Gemini API | |
| # Clean the API key by stripping any extra whitespace or newline characters | |
| gemini_api_key = os.getenv("GEMINI_API_KEY").strip() | |
| # This function will handle the conversation | |
| def chat_with_bot(user_input, history=None): | |
| if history is None: | |
| history = [] # Initialize history if it's None | |
| # Append the user's question to the conversation history | |
| history.append({"role": "user", "content": user_input}) | |
| try: | |
| # Send the conversation history to the Google Gemini API for processing | |
| response = requests.post( | |
| "https://api.gemini.google.com/v1/chat/completions", # Replace with actual Gemini API endpoint | |
| headers={"Authorization": f"Bearer {gemini_api_key}"}, | |
| json={ | |
| "model": "gemini-1", # Replace with the correct Gemini model identifier | |
| "messages": history, | |
| "temperature": 0.7, | |
| "max_tokens": 150, | |
| "top_p": 1, | |
| "frequency_penalty": 0, | |
| "presence_penalty": 0, | |
| }, | |
| ) | |
| # Print out the response status code and body for debugging | |
| print("Status Code:", response.status_code) | |
| print("Response Text:", response.text) | |
| # Try to parse the response JSON | |
| try: | |
| response_data = response.json() # This will throw an error if the response isn't valid JSON | |
| except ValueError: | |
| return f"Error: Invalid JSON response from Gemini API. Response text: {response.text}", history | |
| if response.status_code == 200: | |
| # Extract the response text | |
| bot_response = response_data['choices'][0]['message']['content'] | |
| # Append the bot's response to the conversation history | |
| history.append({"role": "assistant", "content": bot_response}) | |
| # Return both the new bot response and the updated conversation history | |
| return bot_response, history | |
| else: | |
| # Handle errors if status code isn't 200 | |
| return f"Error: {response_data.get('error', {}).get('message', 'Unknown error')}", history | |
| except Exception as e: | |
| # If there's an error with the API request, return an error message | |
| return f"Error: {str(e)}", history | |
| # Define the Gradio interface with a Submit button | |
| chatbot_interface = gr.Interface( | |
| fn=chat_with_bot, | |
| inputs=[ | |
| gr.Textbox(label="Ask me anything", placeholder="Type your question here..."), | |
| gr.State() # The history state input | |
| ], | |
| outputs=[ | |
| gr.Textbox(label="Bot Response", interactive=False), # Output is non-interactive | |
| gr.State() # The history state output | |
| ], | |
| live=False, # Disable live mode so the user submits after typing | |
| title="AI-Powered Multi-Step Q&A Chatbot", | |
| description="This chatbot can handle multi-step conversations. Ask any question and have a conversation!", | |
| flagging_mode="never" # Use the new parameter to disable flagging | |
| ) | |
| # Launch the app | |
| chatbot_interface.launch() # No need for share=True in Hugging Face Spaces | |