multiple files

app.py

@@ -35,17 +35,23 @@ def generate_response(
     max_tokens: int,
     temperature: float,
     top_p: float,
-    file=None
+    files=None
 ) -> Iterator[str]:
     """Generate streaming response from the model"""
     client = get_client()
     
     # Process file if uploaded
-    file_content = process_file(file) if file else ""
+        # Process multiple files if uploaded
+    all_content = ""
+    if files:
+        file_contents = [process_file(file) for file in files]
+        all_content = "\n\n".join([
+            f"File {i+1} content:\n{content}" 
+            for i, content in enumerate(file_contents)
+        ])
     
-    # If there's file content, append it to the message
-    if file_content:
-        message = f"File content:\n{file_content}\n\nUser message:\n{message}"
+    if all_content:
+        message = f"{all_content}\n\nUser message:\n{message}"
 
     messages = [{"role": "system", "content": system_message}]
 
@@ -118,7 +124,8 @@ def main():
         uploaded_file = st.file_uploader(
             "Upload File (optional)",
                type=['txt', 'py', 'md', 'swift', 'java', 'js', 'ts', 'rb', 'go', 
-                'php', 'c', 'cpp', 'h', 'hpp', 'cs', 'html', 'css', 'kt']
+                'php', 'c', 'cpp', 'h', 'hpp', 'cs', 'html', 'css', 'kt'],
+                accept_multiple_files=True  # Add this parameter
         )
 
     # Display chat messages