Update app.py

app.py

@@ -187,14 +187,20 @@ def chat_with_file_contents(prompt, file_content, model_choice='gpt-3.5-turbo'):
     response = openai.ChatCompletion.create(model=model_choice, messages=conversation)
     return response['choices'][0]['message']['content']
 
-def extract_mime_type(file_str):
-    # Using regex pattern matching to find the mime type
-    pattern = r"type='(.*?)'"
-    match = re.search(pattern, file_str)
-    if match:
-        return match.group(1)
+def extract_mime_type(file):
+    # Check if the input is a string
+    if isinstance(file, str):
+        pattern = r"type='(.*?)'"
+        match = re.search(pattern, file)
+        if match:
+            return match.group(1)
+        else:
+            raise ValueError(f"Unable to extract MIME type from {file}")
+    # If it's not a string, assume it's a streamlit.UploadedFile object
+    elif isinstance(file, streamlit.UploadedFile):
+        return file.type
     else:
-        raise ValueError(f"Unable to extract MIME type from {file_str}")
+        raise TypeError("Input should be a string or a streamlit.UploadedFile object")
         
 def pdf2txt(pdf_docs):
     st.write(pdf_docs)