Update app.py

app.py

@@ -50,15 +50,17 @@ def evaluate_pci_compliance(audit_data):
     compliance_analysis = respond(audit_data, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
     return compliance_analysis
 
+# Analyze JSON file input
 def analyze_json_file(file_obj):
-    # Assuming file_obj is a file-like object from Gradio
-    try:
-        # Read and decode the file if it's a file-like object
-        file_str = file_obj.read().decode("utf-8")
-    except AttributeError:
-        # Handle if it's a byte string directly
+    # Handle both file-like objects and strings properly
+    if isinstance(file_obj, bytes):
         file_str = file_obj.decode("utf-8")
-        
+    elif isinstance(file_obj, str):
+        file_str = file_obj
+    else:
+        # Assuming it's a file-like object, read the content as a string
+        file_str = file_obj.read().decode("utf-8")
+    
     json_data = json.loads(file_str)
     audit_data = json.dumps(json_data, indent=2)  # Convert JSON to a formatted string
     return audit_data