Update app.py

app.py

@@ -17,13 +17,7 @@ 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):
@@ -31,36 +25,28 @@ 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
 
-# Streamlit UI
+# Function to generate sign avatar video
+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}"}]})
+        if response.status_code == 200:
+            return "https://example.com/avatar_video.mp4"  # Placeholder 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 = st.tabs([
-        "📸 **Image Load**",
-        "📷 **Take Picture**",
-        "🎥 **Live**"
-    ])
+    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"])
@@ -82,7 +68,6 @@ 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:
@@ -94,6 +79,15 @@ def main():
                 time.sleep(5)
             cap.release()
 
+    with tab4:
+        text_input = st.text_area("Enter text (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)
+            else:
+                st.error("Unable to generate sign video. Please try again.")
+
     with st.sidebar:
         st.markdown("<h2 style='font-size:28px; font-weight: bold; color: #4CAF50;'>Menu</h2>", unsafe_allow_html=True)
         if st.button("📖 About Us", use_container_width=True):
@@ -109,5 +103,4 @@ 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__":
-    main()
+if __name__ == "__main__":