Update app.py

app.py

@@ -1,13 +1,59 @@
 import os
+import sys
+import subprocess
 import streamlit as st
-from io import BytesIO
-from groq import Groq
 
-# Set up the Groq client with your API key directly
-client = Groq(api_key="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")
+# Check if 'groq' is installed, and if not, prompt the user to install it
+def check_groq_installed():
+    try:
+        import groq
+        return True
+    except ImportError:
+        return False
+
+# Check Python version compatibility
+def check_python_version():
+    required_version = (3, 6)  # Minimum required Python version
+    current_version = sys.version_info
+    if current_version < required_version:
+        return False
+    return True
+
+# Check if the Groq client is correctly initialized
+def check_groq_client():
+    try:
+        from groq import Groq
+        # Try initializing the Groq client with the provided API key
+        client = Groq(api_key="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")
+        return True
+    except Exception as e:
+        print(f"Error initializing Groq client: {e}")
+        return False
+
+# Run all checks before proceeding
+def run_checks():
+    # Check if groq is installed
+    if not check_groq_installed():
+        st.error("Groq module is not installed. Please install it using 'pip install groq'.")
+        return False
+    
+    # Check if Python version is compatible
+    if not check_python_version():
+        st.error("Python version 3.6 or higher is required. Please upgrade your Python.")
+        return False
+    
+    # Check if the Groq client can be initialized
+    if not check_groq_client():
+        st.error("Failed to initialize Groq client. Check your API key and the Groq module installation.")
+        return False
+    
+    return True
 
 # Function to request pattern generation from Groq API
 def get_pattern_from_groq(body_measurements, garment_type):
+    from groq import Groq
+    client = Groq(api_key="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")
+
     # Preparing the prompt for the LLM
     prompt = f"Generate a 2D {garment_type} pattern based on the following body measurements:\n{body_measurements}"
     
@@ -30,6 +76,10 @@ def generate_garment_pattern(body_measurements, garment_type):
 
 # Main Streamlit app UI
 def main():
+    # Perform pre-execution checks
+    if not run_checks():
+        return
+
     st.title("Garment Pattern Design Tool")
     
     # Select garment type
@@ -64,22 +114,4 @@ def main():
             pattern_output = generate_garment_pattern(body_measurements_str, garment_type)
             
             # Display the result
-            st.subheader(f"Generated 2D {garment_type} Pattern:")
-            st.text(pattern_output)
-            
-            # Simulating file download
-            buffer = BytesIO()
-            buffer.write(pattern_output.encode())
-            buffer.seek(0)
-            
-            st.download_button(
-                label="Download Garment Pattern",
-                data=buffer,
-                file_name=f"{garment_type.lower().replace(' ', '_')}_pattern.txt",
-                mime="text/plain"
-            )
-        else:
-            st.error("Please enter valid measurements for all fields.")
-    
-if __name__ == "__main__":
-    main()
+