Update app.py

app.py

@@ -102,24 +102,24 @@ def main():
     st.title("API Spec Uploader and Python Interface Generator")
 
     # Upload API Spec File
-    uploaded_file = st.file_uploader("Upload API Spec (JSON, YAML, PDF, DOC)", type=["json", "yaml", "pdf", "docx"])
-
-    if uploaded_file is not None:
-        file_type = uploaded_file.type
-
-        if file_type == "application/json":
-            api_spec_content = uploaded_file.read().decode("utf-8")
-        elif file_type == "application/x-yaml":
-            api_spec_content = uploaded_file.read().decode("utf-8")
-        elif file_type == "application/pdf":
-            # Extract text from PDF
-            api_spec_content = read_pdf(uploaded_file)
-        elif file_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
-            # Extract text from DOCX
-            api_spec_content = read_docx(uploaded_file)
-        else:
-            st.error("Unsupported file format.")
-            return
+        uploaded_file = st.file_uploader("Upload API Spec (JSON, YAML, PDF, DOC)", type=["json", "yaml", "pdf", "docx"])
+
+        if uploaded_file is not None:
+            file_type = uploaded_file.type
+        
+            # Handle different file types
+            if file_type == "application/json":
+                api_spec_content = uploaded_file.read().decode("utf-8")
+            elif file_type == "application/x-yaml":
+                api_spec_content = uploaded_file.read().decode("utf-8")
+            elif file_type == "application/pdf":
+                api_spec_content = read_pdf(uploaded_file)
+            elif file_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document" or file_type == "application/octet-stream":
+                # Handle docx or unknown types
+                api_spec_content = read_docx(uploaded_file)
+            else:
+                st.error(f"Unsupported file format: {file_type}")
+                return
 
         # Display API Spec
         st.subheader("Uploaded API Specification")