Update app.py

app.py

@@ -17,7 +17,13 @@ GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
 
 # Placeholder sign labels
-sign_labels = {0: "Hello", 1: "Thank You", 2: "Yes", 3: "No", 4: "Please"}
+sign_labels = {
+    0: "Hello",
+    1: "Thank You",
+    2: "Yes",
+    3: "No",
+    4: "Please"
+}
 
 # Function to classify sign and refine using Groq API
 def classify_sign(image):
@@ -25,28 +31,58 @@ def classify_sign(image):
     inputs = processor(images=image, return_tensors="pt")
     prediction = inputs['pixel_values'].argmax().item()
     gesture = sign_labels.get(prediction % len(sign_labels), "Unknown Sign")
+
     if GROQ_API_KEY:
-        response = requests.post(GROQ_API_URL, headers={"Content-Type": "application/json", "Authorization": f"Bearer {GROQ_API_KEY}"},
-            json={"model": "llama-3.3-70b-versatile", "messages": [{"role": "user", "content": f"Refine this detected sign: {gesture}"}]})
+        response = requests.post(
+            GROQ_API_URL,
+            headers={
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {GROQ_API_KEY}"
+            },
+            json={
+                "model": "llama-3.3-70b-versatile",
+                "messages": [{"role": "user", "content": f"Refine this detected sign: {gesture}"}]
+            }
+        )
+
         if response.status_code == 200:
             return response.json()['choices'][0]['message']['content']
+
     return gesture
 
-# Function to generate sign avatar video
+# Function to generate sign video from text
 def generate_sign_video(text):
     if GROQ_API_KEY:
-        response = requests.post(GROQ_API_URL, headers={"Content-Type": "application/json", "Authorization": f"Bearer {GROQ_API_KEY}"},
-            json={"model": "llama-3.3-70b-versatile", "messages": [{"role": "user", "content": f"Generate sign language avatar video for: {text}"}]})
+        response = requests.post(
+            GROQ_API_URL,
+            headers={
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {GROQ_API_KEY}"
+            },
+            json={
+                "model": "llama-3.3-70b-versatile",
+                "messages": [{"role": "user", "content": f"Generate sign language video for: {text}"}]
+            }
+        )
+
         if response.status_code == 200:
-            return "https://example.com/avatar_video.mp4"  # Placeholder URL
+            video_url = response.json().get('video_url', '#')
+            return video_url
+
     return None
 
 # Streamlit UI
+
 def main():
     st.set_page_config(page_title="Sign Language Translator", layout="wide")
     st.markdown("<h1 style='text-align: center; font-size: 40px; font-weight: bold; color: #4CAF50;'>🤟 Sign Language Translator</h1>", unsafe_allow_html=True)
 
-    tab1, tab2, tab3, tab4 = st.tabs(["📸 **Image Load**", "📷 **Take Picture**", "🎥 **Live**", "✍️ **Text2Sign**"])
+    tab1, tab2, tab3, tab4 = st.tabs([
+        "📸 **Image Load**",
+        "📷 **Take Picture**",
+        "🎥 **Live**",
+        "📝 **Text2Sign**"
+    ])
 
     with tab1:
         uploaded_image = st.file_uploader("Upload an image of a hand gesture", type=["png", "jpg", "jpeg"])
@@ -68,6 +104,7 @@ def main():
         if st.button("Enable Cam", key="enable_cam"):
             cap = cv2.VideoCapture(0)
             stframe = st.image([])
+
             while cap.isOpened():
                 ret, frame = cap.read()
                 if not ret:
@@ -80,13 +117,16 @@ def main():
             cap.release()
 
     with tab4:
-        text_input = st.text_area("Enter text (max 200 characters)", max_chars=200)
+        text_input = st.text_area("Enter text to generate sign language (Max 200 characters)", max_chars=200)
         if st.button("Generate Sign"):
-            video_url = generate_sign_video(text_input)
-            if video_url:
-                st.video(video_url)
+            if text_input:
+                video_url = generate_sign_video(text_input)
+                if video_url:
+                    st.video(video_url)
+                else:
+                    st.error("Failed to generate sign language video.")
             else:
-                st.error("Unable to generate sign video. Please try again.")
+                st.warning("Please enter some text.")
 
     with st.sidebar:
         st.markdown("<h2 style='font-size:28px; font-weight: bold; color: #4CAF50;'>Menu</h2>", unsafe_allow_html=True)
@@ -103,4 +143,5 @@ def main():
         if st.button("💬 Feedback", use_container_width=True):
             st.text_area("We value your feedback! Please share your thoughts below:")
 
-if __name__ == "__main__":
+if __name__ == "__main__":
+    main()