Spaces:
Sleeping
Sleeping
import gradio as gr | |
import pathlib | |
import textwrap | |
import google.generativeai as genai | |
def to_markdown(text): | |
"""Converts text to Markdown format with proper indentation.""" | |
text = text.replace('•', ' *') | |
return textwrap.indent(text, '> ', lambda line: True) | |
def chat(message, history, img=None): | |
"""Generates a response to the user's message, optionally using an image.""" | |
genai.configure(api_key='AIzaSyCMBk81YmILNTok8hd6tYtJaevp1qbl6I0') # Replace with your actual API key | |
text_model = genai.GenerativeModel('gemini-pro') | |
#vision_model = genai.GenerativeModel('gemini-pro-vision') | |
try: | |
# Use only text model | |
response = text_model.generate_content(message, stream=True) | |
for chunk in response: | |
return to_markdown(chunk.text) # Format as Markdown | |
except Exception as e: | |
print(f"Error during generation: {e}") | |
return "An error occurred while generating the response. Please try again later." | |
chat_interface = gr.ChatInterface( | |
fn=chat, | |
title="Gemini Chat", | |
description="Chat with an AI assistant powered by Gemini", | |
theme="soft" | |
) | |
chat_interface.launch() | |