Update app.py

app.py

@@ -164,45 +164,57 @@ def unzip_and_process(zip_file, model_choice, openai_classifier, questions_strin
 
 def run_interface(questions_json, resumes_zip):
     try:
-        # Handle the questions.json file
-        if isinstance(questions_json, str):
+        # Handle the questions.json file content
+        if isinstance(questions_json, bytes):
+            # Parse bytes directly as JSON string
+            questions_string = questions_json.decode('utf-8')
+        elif isinstance(questions_json, str):
             # If it's a file path
             with open(questions_json, 'r') as f:
                 questions_string = f.read()
         else:
-            # If it's a temporary file object from Gradio
-            questions_string = questions_json.name
-            with open(questions_string, 'r') as f:
-                questions_string = f.read()
+            raise ValueError("Unexpected questions_json type")
 
         # Handle the ZIP file
-        if isinstance(resumes_zip, str):
+        if isinstance(resumes_zip, bytes):
+            # Write bytes to a temporary zip file
+            temp_zip_path = "temp_uploaded_resumes.zip"
+            with open(temp_zip_path, "wb") as f:
+                f.write(resumes_zip)
+            zip_path = temp_zip_path
+        elif isinstance(resumes_zip, str):
             zip_path = resumes_zip
         else:
-            # If it's a temporary file object from Gradio
-            zip_path = resumes_zip.name
+            raise ValueError("Unexpected resumes_zip type")
 
         openai_classifier = OpenAIClassifier(api_key="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")
         output_file = unzip_and_process(zip_path, "openai", openai_classifier, questions_string)
         
+        # Clean up temporary zip file if it was created
+        if isinstance(resumes_zip, bytes) and os.path.exists(temp_zip_path):
+            os.remove(temp_zip_path)
+            
         return output_file
 
     except Exception as e:
         print(f"Error in run_interface: {str(e)}")
+        # Clean up temporary files in case of error
+        if 'temp_zip_path' in locals() and os.path.exists(temp_zip_path):
+            os.remove(temp_zip_path)
         raise gr.Error(f"Processing failed: {str(e)}")
 
-# Set up the Gradio interface with correct file type parameters
+# Set up the Gradio interface
 interface = gr.Interface(
     fn=run_interface,
     inputs=[
         gr.File(
             label="Upload questions.json", 
-            type="binary",  # Changed from 'file' to 'binary'
+            type="binary",
             file_types=[".json"]
         ),
         gr.File(
             label="Upload ZIP of Resumes", 
-            type="binary",  # Changed from 'file' to 'binary'
+            type="binary",
             file_types=[".zip"]
         )
     ],
@@ -212,5 +224,5 @@ interface = gr.Interface(
     allow_flagging="never",
 )
 
-# Launch with debugging enabled
-interface.launch(debug=True)
+# Launch with debugging and sharing enabled
+interface.launch(debug=True, share=True)