Spaces:
Sleeping
Sleeping
File size: 3,524 Bytes
67c0817 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import gradio as gr
import base64
from groq import Groq
import tempfile
import os
def encode_image(image_path):
"""Convert an image to base64 encoding"""
with open(image_path, "rb") as image_file:
return base64.b64encode(image_file.read()).decode('utf-8')
def extract_medicines(image, api_key):
"""Extract medicine names from prescription image using Groq API"""
if not api_key:
return "Please provide a Groq API key"
if image is None:
return "Please upload a prescription image"
try:
# Save the uploaded image to a temporary file
with tempfile.NamedTemporaryFile(suffix='.jpg', delete=False) as temp_image:
image_path = temp_image.name
image.save(image_path)
# Encode the image to base64
base64_image = encode_image(image_path)
# Clean up the temporary file
os.unlink(image_path)
# Initialize Groq client with the provided API key
client = Groq(api_key=api_key)
# Create the prompt specifically asking for medicine names only
prompt = "This is an image of a medical prescription. Extract and list ONLY the names of medicines/drugs/medications from this prescription. Do not include dosages, frequencies, or other information. Return just a simple list of medicine names, one per line."
# Make the API call
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": [
{"type": "text", "text": prompt},
{
"type": "image_url",
"image_url": {
"url": f"data:image/jpeg;base64,{base64_image}",
},
},
],
}
],
model="meta-llama/llama-4-scout-17b-16e-instruct",
)
# Return the extracted medicine names
return chat_completion.choices[0].message.content
except Exception as e:
return f"An error occurred: {str(e)}"
# Create the Gradio interface
with gr.Blocks(title="Prescription Medicine Extractor") as app:
gr.Markdown("# Medicine Name Extractor from Prescriptions")
gr.Markdown("Upload a prescription image and enter your Groq API key to extract medicine names")
with gr.Row():
with gr.Column():
api_key_input = gr.Textbox(
label="Groq API Key",
placeholder="Enter your Groq API key here",
type="password"
)
image_input = gr.Image(label="Upload Prescription Image", type="pil")
extract_button = gr.Button("Extract Medicine Names")
with gr.Column():
output = gr.Textbox(label="Extracted Medicine Names", lines=10)
extract_button.click(
fn=extract_medicines,
inputs=[image_input, api_key_input],
outputs=output
)
gr.Markdown("""
## Instructions
1. Enter your Groq API key
2. Upload a clear image of a medical prescription
3. Click "Extract Medicine Names"
4. The application will return only the names of medicines from the prescription
**Note:** Your API key is not stored and is only used for making requests to the Groq API.
""")
if __name__ == "__main__":
app.launch() |