Update app.py

app.py

@@ -0,0 +1,64 @@
+import streamlit as st
+from transformers import pipeline
+import requests
+
+# Load OCR model for extracting text from images
+@st.cache_resource
+def load_ocr_model():
+    return pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+
+# Function to interact with ChatGPT API
+def chat_with_gpt(prompt):
+    api_key = st.secrets["OPENAI_API_KEY"]
+    url = "https://api.openai.com/v1/completions"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json",
+    }
+    data = {
+        "model": "text-davinci-003",
+        "prompt": prompt,
+        "max_tokens": 1000,
+        "temperature": 0.7,
+    }
+    response = requests.post(url, headers=headers, json=data)
+    if response.status_code == 200:
+        return response.json().get("choices")[0].get("text", "").strip()
+    else:
+        st.error(f"Error from OpenAI: {response.status_code}")
+        return None
+
+# Streamlit App
+def main():
+    st.title("Image-to-Text with ChatGPT")
+    st.markdown(
+        """
+        **Upload an image**, extract text using a state-of-the-art OCR model, 
+        and get explanations or solutions using ChatGPT.
+        """
+    )
+
+    uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
+
+    if uploaded_file is not None:
+        st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+        st.write("Processing...")
+
+        # Extract text from image
+        ocr_model = load_ocr_model()
+        extracted_text = ocr_model(uploaded_file)["generated_text"]
+
+        if extracted_text:
+            st.write("### Extracted Text:")
+            st.write(extracted_text)
+
+            # Send text to ChatGPT
+            st.write("### ChatGPT Explanation:")
+            explanation = chat_with_gpt(extracted_text)
+            if explanation:
+                st.write(explanation)
+        else:
+            st.error("Could not extract text. Please try again with another image.")
+
+if __name__ == "__main__":
+    main()