import gradio as gr
import google.generativeai as genai

model = genai.GenerativeModel('gemini-pro-vision')
def generate_prescription(image):
    response = model.generate_content(["Write a prescription in pointer format ordered by name of medicine, symptoms, primary diagnosis, usage and dosage of medicine in the image. Make sure to ask person to visit doctor if problem presists.", image])
    return response.text

interface = gr.Interface(fn=generate_prescription,
                    inputs=gr.Image(label="Upload image", type="pil"),
                    outputs=gr.Textbox(label="Your prescription is here:"),
                    title="Medicine Prescription",
                    description="Find Prescription to any medicine",
                    allow_flagging="never",
                    examples=["Combiflam.jpg", "Zintac.jpg"])
interface.launch(debug = True)