Update app.py

app.py

@@ -396,9 +396,14 @@ def handle_message():
 def make_text():
     query = request.args.get('query')
     file_name = request.args.get('fileName')
+    repo_id = request.args.get('repoId')
+    repo_type = request.args.get('repoType', 'dataset')
 
-    if not query or not file_name:
-        return "Both 'query' and 'fileName' parameters are required", 400
+    if not query or not file_name or not repo_id:
+        return "Parameters 'query', 'fileName', and 'repoId' are required", 400
+
+    if repo_type not in ['space', 'dataset', 'model']:
+        return "Invalid 'repoType'. Must be 'space', 'dataset', or 'model'", 400
 
     # Create a temporary file
     with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
@@ -410,10 +415,10 @@ def make_text():
         api.upload_file(
             path_or_fileobj=temp_file_path,
             path_in_repo=file_name,
-            repo_id="thunder-lord/test-1",
-            repo_type="dataset",
+            repo_id=repo_id,
+            repo_type=repo_type,
         )
-        return f"File '{file_name}' uploaded successfully", 200
+        return f"File '{file_name}' uploaded successfully to {repo_id} ({repo_type})", 200
     except Exception as e:
         return f"Error uploading file: {str(e)}", 500
     finally: