Update app.py

app.py

@@ -198,17 +198,6 @@ class ModelManager:
             elif chosen_inference == "litellm":
                 return LiteLLMModel(model_id=model_id)
 
-            elif chosen_inference == "ollama":
-                if not key_manager:
-                    raise ValueError("Key manager required for Ollama model")
-
-                return LiteLLMModel(
-                    model_id=model_id,
-                    api_base="http://localhost:11434",
-                    api_key=key_manager.get_key("ollama_api_key"),
-                    num_ctx=8192,
-                )
-
             elif chosen_inference == "openai":
                 if not key_manager:
                     raise ValueError("Key manager required for OpenAI model")
@@ -397,6 +386,8 @@ class GradioUI:
             "application/pdf",
             "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
             "text/plain",
+            "image/png",  # Add PNG MIME type
+            "video/mp4",  # Add MP4 MIME type"
         ],
     ):
         """Handle file uploads with proper validation and security."""
@@ -685,4 +676,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-