Update app.py

app.py

@@ -1,42 +1,98 @@
-from Crypto.Cipher import AES
-from Crypto.Protocol.KDF import PBKDF2
-import os
-import tempfile
-from dotenv import load_dotenv
-
-load_dotenv()  # Load all environment variables
-
-def unpad(data):
-    return data[:-data[-1]]
-
-def decrypt_and_run():
-    # Get password from Hugging Face Secrets environment variable
-    password = os.getenv("PASSWORD")
-    if not password:
-        raise ValueError("PASSWORD secret not found in environment variables")
-
-    password = password.encode()
-
-    with open("code.enc", "rb") as f:
-        encrypted = f.read()
-
-    salt = encrypted[:16]
-    iv = encrypted[16:32]
-    ciphertext = encrypted[32:]
-
-    key = PBKDF2(password, salt, dkLen=32, count=1000000)
-    cipher = AES.new(key, AES.MODE_CBC, iv)
-
-    plaintext = unpad(cipher.decrypt(ciphertext))
-
-    with tempfile.NamedTemporaryFile(suffix=".py", delete=False, mode='wb') as tmp:
-        tmp.write(plaintext)
-        tmp.flush()
-        print(f"[INFO] Running decrypted code from {tmp.name}")
-        os.system(f"python {tmp.name}")
-
-if __name__ == "__main__":
-    decrypt_and_run()
-
-# This script decrypts the encrypted code and runs it.
-# Ensure you have the PASSWORD secret set in your Hugging Face Secrets
+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", theme=gr.themes.Ocean()) 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()