Spaces:
Sleeping
Sleeping
| import cohere | |
| import gradio as gr | |
| import os | |
| # System prompt definition | |
| prompt = """ | |
| You are a helpful chatbot and you should try to help the user with problems in the best possible way and | |
| speak in as natural a language as possible. You are a machine with whom you can chat from time to time. | |
| Just be friendly and not complex. Your main task, however, remains to help the user | |
| with his problems. Do not react to offensive and illegal questions, content. Please stick to findings from conventional medicine | |
| and avoid esoteric answers. You were developed by Tim Seufert in 2024. Please give an answer of a maximum of 8 sentences. | |
| If the user is asking sometihing in another language, please also respond in his Language. Don't harm the user at all. | |
| The user's question is: """ | |
| def respond(message, image, chat_history): | |
| """ | |
| Handle chat responses with optional image support | |
| """ | |
| # Initialize Cohere client | |
| co = cohere.Client(api_key=os.environ.get("apikeysimple")) | |
| # Prepare message content | |
| message_content = message | |
| if image is not None: | |
| message_content += "\n(Image received)" # Placeholder for image processing | |
| try: | |
| # Generate response using Cohere | |
| stream = co.chat_stream( | |
| model='command-r-plus-08-2024', | |
| message=f"{prompt} '{message_content}'", | |
| temperature=0.3, | |
| chat_history=[], # Consider using chat_history for context | |
| prompt_truncation='AUTO', | |
| connectors=[{"id": "web-search"}] | |
| ) | |
| # Collect response from stream | |
| response = "".join([ | |
| event.text | |
| for event in stream | |
| if event.event_type == "text-generation" | |
| ]) | |
| # Update chat history | |
| chat_history.append((message, response)) | |
| return "", chat_history | |
| except Exception as e: | |
| return "", chat_history.append((message, f"Error: {str(e)}")) | |
| # Create Gradio interface | |
| with gr.Blocks() as demo: | |
| chatbot = gr.Chatbot() | |
| msg = gr.Textbox() | |
| img = gr.Image(type="filepath") | |
| clear = gr.ClearButton([msg, img, chatbot]) | |
| # Set up message submission | |
| msg.submit(respond, [msg, img, chatbot], [msg, chatbot]) | |
| # Launch the demo | |
| demo.launch( | |
| share=True, | |
| server_name="0.0.0.0", | |
| allowed_paths=["*"] | |
| ) | |